| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 655 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
690 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
723 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
759 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
961 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
989 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1019 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1047 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1080 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1113 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 | 165 TEST_F(HloCostAnalysisTest, DotGeneral) { in TEST_F() argument
176 DotGeneral(lhs, rhs, dnums); in TEST_F()
212 DotGeneral(lhs, rhs, dnums); in TEST_F()
242 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 | 1303 TEST_F(ShapeInferenceTest, DotGeneral) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1373 def DotGeneral(self, lhs, rhs, dimension_numbers, precision_config=None): member in ComputationBuilder
1387 return ops.DotGeneral(
|
| D | xla_client_test.py | 760 c.DotGeneral(c.Constant(lhs), c.Constant(rhs), dimension_numbers)
775 c.DotGeneral(c.Constant(lhs), c.Constant(rhs), dimension_numbers)
787 c.DotGeneral(
|
| D | xla.cc | 857 ops.def("DotGeneral", &DotGeneral, py::arg("lhs"), py::arg("rhs"), in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 427 XlaOp DotGeneral(XlaOp lhs, XlaOp rhs,
820 friend XlaOp DotGeneral(XlaOp lhs, XlaOp rhs,
1389 XlaOp DotGeneral(XlaOp lhs, XlaOp rhs,
|
| D | xla_builder.cc | 1143 return DotGeneral(lhs, rhs, dimension_numbers, precision_config); in Dot()
1147 XlaOp XlaBuilder::DotGeneral(XlaOp lhs, XlaOp rhs, in DotGeneral() function in xla::XlaBuilder
3028 XlaOp DotGeneral(const XlaOp lhs, const XlaOp rhs, in DotGeneral() function
3031 return lhs.builder()->DotGeneral(lhs, rhs, dimension_numbers, in DotGeneral()
|
| D | xla_builder_test.cc | 748 DotGeneral(lhs, rhs, dnums); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 495 auto dot = DotGeneral(x, y, dnums, &precision_proto); in Einsum()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 942 ## DotGeneral section in Operation Semantics
945 [`XlaBuilder::DotGeneral`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
947 <b> `DotGeneral(lhs, rhs, dimension_numbers)` </b>
965 DotGeneral performs the sum of products over contracting dimensions specified
984 DotGeneral(lhs, rhs, dnums) -> { {6.0, 12.0},
1010 DotGeneral(lhs, rhs, dnums) -> { { {1.0, 2.0},
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 212 xla::DotGeneral(p0, p1, ddn); in Dot()
|