| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 620 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
657 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
691 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
728 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
930 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
958 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
988 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1016 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1049 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1082 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 410 auto bottom_right_minus_bottom_left = xla::DotGeneral( in CalculateGradWarp()
418 auto top_right_minus_top_left = xla::DotGeneral( in CalculateGradWarp()
426 auto bottom_right_minus_top_right = xla::DotGeneral( in CalculateGradWarp()
434 auto bottom_left_minus_top_left = xla::DotGeneral( in CalculateGradWarp()
547 auto blended_pixels = xla::DotGeneral(weights, neighbors_data, dot_dims, in Compile()
|
| D | xla_dot_op.cc | 50 xla::XlaOp output = xla::DotGeneral(context->Input(0), context->Input(1), in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 161 TEST_F(HloCostAnalysisTest, DotGeneral) { in TEST_F() argument
172 DotGeneral(lhs, rhs, dnums); in TEST_F()
201 DotGeneral(lhs, rhs, dnums); in TEST_F()
224 DotGeneral(lhs, rhs, dnums); in TEST_F()
|
| D | triangular_solve_expander.cc | 229 auto update = -DotGeneral(input_row, body_out, dnums, &precision_proto); in InvertDiagonalBlocks()
|
| D | shape_inference_test.cc | 1048 TEST_F(ShapeInferenceTest, DotGeneral) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 251 return Transpose(DotGeneral(x, y, dnums, &precision_proto), transpose_dims); in Einsum()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 580 LocalOp ComputationBuilder::DotGeneral( in DotGeneral() function in xla::swig::ComputationBuilder
583 return xla::DotGeneral(lhs.op(), rhs.op(), dimension_numbers); in DotGeneral()
|
| D | local_computation_builder.h | 290 LocalOp DotGeneral(const LocalOp& lhs, const LocalOp& rhs,
|
| D | local_computation_builder.i | 338 %unignore xla::swig::ComputationBuilder::DotGeneral;
|
| D | xla_client.py | 1604 def DotGeneral(self, lhs, rhs, dimension_numbers): member in ComputationBuilder
1618 return self._client.DotGeneral(lhs, rhs, dimension_numbers)
|
| D | xla_client_test.py | 660 c.DotGeneral(c.Constant(lhs), c.Constant(rhs), dimension_numbers)
675 c.DotGeneral(c.Constant(lhs), c.Constant(rhs), dimension_numbers)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 379 XlaOp DotGeneral(const XlaOp& lhs, const XlaOp& rhs,
778 friend XlaOp DotGeneral(const XlaOp& lhs, const XlaOp& rhs,
1298 XlaOp DotGeneral(const XlaOp& lhs, const XlaOp& rhs,
|
| D | xla_builder.cc | 1019 return DotGeneral(lhs, rhs, dimension_numbers, precision_config); in Dot()
1023 XlaOp XlaBuilder::DotGeneral(const XlaOp& lhs, const XlaOp& rhs, in DotGeneral() function in xla::XlaBuilder
2963 XlaOp DotGeneral(const XlaOp& lhs, const XlaOp& rhs, in DotGeneral() function
2966 return lhs.builder()->DotGeneral(lhs, rhs, dimension_numbers, in DotGeneral()
|
| D | xla_builder_test.cc | 738 DotGeneral(lhs, rhs, dnums); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 906 ## DotGeneral section in Operation Semantics
909 [`XlaBuilder::DotGeneral`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
911 <b> `DotGeneral(lhs, rhs, dimension_numbers)` </b>
929 DotGeneral performs the sum of products over contracting dimensions specified
948 DotGeneral(lhs, rhs, dnums) -> { {6.0, 12.0},
974 DotGeneral(lhs, rhs, dnums) -> { { {1.0, 2.0},
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 181 xla::DotGeneral(p0, p1, ddn); in Dot()
|