Searched refs:rhs_literal (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_evaluator.cc | 62 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 …]
|
D | hlo_evaluator_typed_visitor.h | 1088 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 …]
|
D | hlo_evaluator_test.cc | 807 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/ |
D | reference_util.cc | 416 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/ |
D | array_elementwise_ops_test.cc | 275 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()
|