Home
last modified time | relevance | path

Searched refs:lhs_constant (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc823 auto lhs_constant = ConstantR2FromArray2D(&builder, *constant_lhs_array); in XLA_TYPED_TEST() local
830 Dot(lhs_constant, in XLA_TYPED_TEST()
921 auto lhs_constant = ConstantR2FromArray2D(&builder, *constant_lhs_array); in XLA_TEST_F() local
925 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
949 auto lhs_constant = ConstantR2FromArray2D(&builder, *constant_lhs_array); in XLA_TEST_F() local
958 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
979 auto lhs_constant = ConstantR2FromArray2D(&builder, *constant_lhs_array); in XLA_TEST_F() local
983 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1007 auto lhs_constant = ConstantR2FromArray2D(&builder, *constant_lhs_array); in XLA_TEST_F() local
1016 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators_test.cc69 for (const XlaOp& lhs_constant : all_constants) { in BuildComparatorAndComparisons() local
72 Call(test->builder(), compare, {lhs_constant, rhs_constant}), {1})); in BuildComparatorAndComparisons()
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc1134 if (auto* lhs_constant = dynamic_cast<ConstantArray*>(lhs)) { in ComputeArrayForDot() local
1136 precision_config, lhs_constant, in ComputeArrayForDot()