Home
last modified time | relevance | path

Searched refs:rhs_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
99 rhs_literal.Get<OperandT>(multi_index)); in Compare()
109 LiteralSlice rhs_literal) { in Compare() argument
131 rhs_literal.Get<complex64>(multi_index)); in Compare()
141 LiteralSlice rhs_literal) { in Compare() argument
163 rhs_literal.Get<complex128>(multi_index)); in Compare()
682 const Literal& rhs_literal = GetEvaluatedLiteralFor(rhs); 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.cc730 auto rhs_literal = LiteralUtil::CreateR2<float>({{1, 2}}); in TEST_P() local
732 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
776 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P() local
778 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
820 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P() local
822 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
857 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*rhs_array); in TEST_P() local
859 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
900 auto rhs_literal = LiteralUtil::CreateR3FromArray3D<float>(rhs_array); in TEST_P() local
902 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
[all …]
Dhlo_evaluator_typed_visitor.h1047 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in HandleConvolution() local
1061 auto rhs_literal_data = rhs_literal.data<ReturnT>(); in HandleConvolution()
1250 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in HandleDot() local
1258 rhs_array.SetValues(rhs_literal.data<NativeT>()); in HandleDot()
1291 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in HandleDotSlowPath() local
1366 static_cast<ElementwiseT>(rhs_literal.Get<ReturnT>(rhs_index)); in HandleDotSlowPath()
2954 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in ElementWiseBinaryOp() local
2962 rhs_literal.Get<ReturnT>(multi_index)); in ElementWiseBinaryOp()
2980 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in ElementwiseTernaryOp() local
2988 rhs_literal.Get<RhsType>(multi_index), in ElementwiseTernaryOp()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc482 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs); in ConvArray4DGeneralDimensionsDilated() local
500 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
502 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
513 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(0))); in ConvArray4DGeneralDimensionsDilated()
523 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(1))); in ConvArray4DGeneralDimensionsDilated()
533 lhs_literal.shape(), rhs_literal.shape(), in ConvArray4DGeneralDimensionsDilated()
540 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc242 Literal rhs_literal = LiteralUtil::CreateR1<uint64>({rhs}); in XLA_TEST_F() local
243 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F()
245 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
281 Literal rhs_literal = LiteralUtil::CreateR1<int64>({rhs}); in XLA_TEST_F() local
282 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F()
284 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
304 Literal rhs_literal = LiteralUtil::CreateR1<uint64>({rhs}); in XLA_TEST_F() local
305 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F()
309 ComputeAndCompare(&b, {std::move(lhs_literal), std::move(rhs_literal)}); in XLA_TEST_F()