| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 664 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
699 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
732 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
768 DotGeneral(x, y, dnums); in XLA_TYPED_TEST()
970 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
998 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1028 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1056 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1089 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1122 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 | 350 auto update = -DotGeneral(input_row, body_out, dnums, &precision_proto); in InvertDiagonalBlocks()
|
| D | shape_inference_test.cc | 1559 TEST_F(ShapeInferenceTest, DotGeneral) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 148 ops.def("DotGeneral", &DotGeneral, py::arg("lhs"), py::arg("rhs"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 713 ops.DotGeneral(
729 ops.DotGeneral(
743 ops.DotGeneral(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 836 DotGeneral(lhs, rhs, dnums); in TEST_F()
1093 DotGeneral(p0, p1, dnums, /*precision_config=*/nullptr, in TEST_F()
|
| D | xla_builder.h | 528 XlaOp DotGeneral(
1122 friend XlaOp DotGeneral(XlaOp lhs, XlaOp rhs,
1862 XlaOp DotGeneral(
|
| D | xla_builder.cc | 1295 return DotGeneral(lhs, rhs, dimension_numbers, precision_config); in Dot()
1299 XlaOp XlaBuilder::DotGeneral( in DotGeneral() function in xla::XlaBuilder
4304 XlaOp DotGeneral(const XlaOp lhs, const XlaOp rhs, in DotGeneral() function
4308 return lhs.builder()->DotGeneral(lhs, rhs, dimension_numbers, in DotGeneral()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 514 auto dot = DotGeneral(x, y, dnums, &precision_proto); in Einsum()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 993 ## DotGeneral section in Operation Semantics
996 [`XlaBuilder::DotGeneral`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
998 <b> `DotGeneral(lhs, rhs, dimension_numbers)` </b>
1016 DotGeneral performs the sum of products over contracting dimensions specified
1035 DotGeneral(lhs, rhs, dnums) -> { {6.0, 12.0},
1061 DotGeneral(lhs, rhs, dnums) -> { { {1.0, 2.0},
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 297 xla::DotGeneral(p0, p1, ddn); in Dot()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17686 let summary = "Wraps the XLA DotGeneral operator, documented at";
|