Searched refs:DotGeneral (Results 1 - 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
H A D | batch_dot.cc | 114 return builder->DotGeneral(x, y, dot_dnums);
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
H A D | local_computation_builder.cc | 418 ComputationDataHandle LocalComputationBuilder::DotGeneral( function in class:xla::swig::LocalComputationBuilder 421 return builder_.DotGeneral(lhs, rhs, dimension_numbers);
|
H A D | local_computation_builder.h | 203 ComputationDataHandle DotGeneral(
|
H A D | xla_client.py | 1001 def DotGeneral(self, lhs, rhs, dimension_numbers): member in class:ComputationBuilder 1012 Returns: a ComputationDataHandle representing the DotGeneral operation. 1017 self._client.DotGeneral(
|
H A D | local_computation_builder.i | 916 %unignore xla::swig::LocalComputationBuilder::DotGeneral;
|
H A D | xla_client_test.py | 454 c.DotGeneral(c.Constant(lhs), c.Constant(rhs), dimension_numbers) 469 c.DotGeneral(c.Constant(lhs), c.Constant(rhs), dimension_numbers)
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | dot_operation_test.cc | 540 auto out = builder.DotGeneral(x, y, dnums);
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | computation_builder.cc | 544 return DotGeneral(lhs, rhs, dimension_numbers); 547 ComputationDataHandle ComputationBuilder::DotGeneral( function in class:xla::ComputationBuilder
|
H A D | computation_builder.h | 347 ComputationDataHandle DotGeneral(
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | shape_inference_test.cc | 985 TEST_F(ShapeInferenceTest, DotGeneral) {
|
Completed in 140 milliseconds