Home
last modified time | relevance | path

Searched refs:lhs_literal (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc60 LiteralSlice lhs_literal, LiteralSlice rhs_literal) { in Compare() argument
98 return compare_op(lhs_literal.Get<OperandT>(multi_index), in Compare()
108 LiteralSlice lhs_literal, in Compare() argument
130 return compare_op(lhs_literal.Get<complex64>(multi_index), in Compare()
140 LiteralSlice lhs_literal, in Compare() argument
162 return compare_op(lhs_literal.Get<complex128>(multi_index), in Compare()
681 const Literal& lhs_literal = GetEvaluatedLiteralFor(lhs); in HandleCompare() local
689 Compare<bool>(compare->shape(), direction, lhs_literal, rhs_literal)); in HandleCompare()
694 lhs_literal, rhs_literal)); in HandleCompare()
699 lhs_literal, rhs_literal)); in HandleCompare()
[all …]
Dhlo_evaluator_test.cc724 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P() local
726 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
764 auto lhs_literal = LiteralUtil::CreateR1<float>({1, 2, 3}); in TEST_P() local
766 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
808 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P() local
810 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
851 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); in TEST_P() local
853 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
895 auto lhs_literal = LiteralUtil::CreateR3FromArray3D<float>(lhs_array); in TEST_P() local
897 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
[all …]
Dhlo_evaluator_typed_visitor.h1046 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in HandleConvolution() local
1060 auto lhs_literal_data = lhs_literal.data<ReturnT>(); in HandleConvolution()
1249 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in HandleDot() local
1255 lhs_array.SetValues(lhs_literal.data<NativeT>()); in HandleDot()
1290 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in HandleDotSlowPath() local
1365 static_cast<ElementwiseT>(lhs_literal.Get<ReturnT>(lhs_index)) * in HandleDotSlowPath()
2953 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in ElementWiseBinaryOp() local
2961 lhs_literal.Get<ReturnT>(multi_index), in ElementWiseBinaryOp()
2979 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in ElementwiseTernaryOp() local
2987 return ternary_op(lhs_literal.Get<LhsType>(multi_index), in ElementwiseTernaryOp()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc481 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs); in ConvArray4DGeneralDimensionsDilated() local
496 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
498 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
533 lhs_literal.shape(), rhs_literal.shape(), in ConvArray4DGeneralDimensionsDilated()
538 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc228 Literal lhs_literal = LiteralUtil::CreateR1<uint64>({lhs}); in XLA_TEST_F() local
229 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param"); in XLA_TEST_F()
231 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
268 Literal lhs_literal = LiteralUtil::CreateR1<int64>({lhs}); in XLA_TEST_F() local
269 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param"); in XLA_TEST_F()
271 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
300 Literal lhs_literal = LiteralUtil::CreateR1<uint64>({lhs}); in XLA_TEST_F() local
301 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param"); in XLA_TEST_F()
309 ComputeAndCompare(&b, {std::move(lhs_literal), std::move(rhs_literal)}); in XLA_TEST_F()