Searched refs:rhs_constant (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 871 auto rhs_constant = ConstantR2FromArray2D(&builder, *constant_rhs_array); in XLA_TYPED_TEST() local
879 rhs_constant); in XLA_TYPED_TEST()
922 auto rhs_constant = ConstantR2FromArray2D(&builder, *constant_rhs_array); in XLA_TEST_F() local
930 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
950 auto rhs_constant = ConstantR2FromArray2D(&builder, *constant_rhs_array); in XLA_TEST_F() local
953 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
980 auto rhs_constant = ConstantR2FromArray2D(&builder, *constant_rhs_array); in XLA_TEST_F() local
988 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1008 auto rhs_constant = ConstantR2FromArray2D(&builder, *constant_rhs_array); in XLA_TEST_F() local
1011 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators_test.cc | 70 for (const XlaOp& rhs_constant : all_constants) { in BuildComparatorAndComparisons() local
72 Call(test->builder(), compare, {lhs_constant, rhs_constant}), {1})); in BuildComparatorAndComparisons()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 1125 if (auto* rhs_constant = dynamic_cast<ConstantArray*>(rhs)) { in ComputeArrayForDot() local
1128 lhs_indexed_array, rhs_constant); in ComputeArrayForDot()
|