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.cc62 LiteralSlice lhs_literal, LiteralSlice rhs_literal) { in Compare() argument
101 rhs_literal.Get<OperandT>(multi_index)); in Compare()
111 LiteralSlice rhs_literal) { in Compare() argument
133 rhs_literal.Get<complex64>(multi_index)); in Compare()
143 LiteralSlice rhs_literal) { in Compare() argument
165 rhs_literal.Get<complex128>(multi_index)); in Compare()
719 const Literal& rhs_literal = GetEvaluatedLiteralFor(rhs); in HandleCompare() local
726 Compare<bool>(compare->shape(), direction, lhs_literal, rhs_literal)); in HandleCompare()
731 lhs_literal, rhs_literal)); in HandleCompare()
736 lhs_literal, rhs_literal)); in HandleCompare()
[all …]
Dhlo_evaluator_typed_visitor.h1088 const Literal& rhs_literal) {
1092 const Shape& rhs_shape = rhs_literal.shape();
1120 auto rhs_literal_data = rhs_literal.data<ReturnT>();
1301 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs);
1305 return HandleConvolutionWithLiterals(conv, lhs_literal, rhs_literal);
1310 rhs_literal);
1315 rhs_literal.Convert(result_shape.element_type()).ValueOrDie());
1319 rhs_literal.Convert(result_shape.element_type()).ValueOrDie());
1370 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs);
1378 rhs_array.SetValues(rhs_literal.data<NativeT>());
[all …]
Dhlo_evaluator_test.cc807 auto rhs_literal = LiteralUtil::CreateR2<float>({{1, 2}}); in TEST_P() local
809 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
853 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P() local
855 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
897 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P() local
899 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
934 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*rhs_array); in TEST_P() local
936 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
977 auto rhs_literal = LiteralUtil::CreateR3FromArray3D<float>(rhs_array); in TEST_P() local
979 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc416 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs); in ConvArray4DGeneralDimensionsDilated() local
434 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
436 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
447 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(0))); in ConvArray4DGeneralDimensionsDilated()
457 rhs_literal.shape().dimensions(dnums.kernel_spatial_dimensions(1))); in ConvArray4DGeneralDimensionsDilated()
467 lhs_literal.shape(), rhs_literal.shape(), in ConvArray4DGeneralDimensionsDilated()
475 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal))); in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc275 Literal rhs_literal = LiteralUtil::CreateR1<uint64>({rhs}); in XLA_TEST_F() local
276 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F()
278 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
314 Literal rhs_literal = LiteralUtil::CreateR1<int64>({rhs}); in XLA_TEST_F() local
315 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F()
317 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
337 Literal rhs_literal = LiteralUtil::CreateR1<uint64>({rhs}); in XLA_TEST_F() local
338 auto rhs_param = Parameter(&b, 1, rhs_literal.shape(), "rhs_param"); in XLA_TEST_F()
342 ComputeAndCompare(&b, {std::move(lhs_literal), std::move(rhs_literal)}); in XLA_TEST_F()