Searched refs:lhs_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
100 return compare_op(lhs_literal.Get<OperandT>(multi_index), in Compare()
110 LiteralSlice lhs_literal, in Compare() argument
132 return compare_op(lhs_literal.Get<complex64>(multi_index), in Compare()
142 LiteralSlice lhs_literal, in Compare() argument
164 return compare_op(lhs_literal.Get<complex128>(multi_index), in Compare()
718 const Literal& lhs_literal = GetEvaluatedLiteralFor(lhs); 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 | 1087 const Literal& lhs_literal,
1091 const Shape& lhs_shape = lhs_literal.shape();
1119 auto lhs_literal_data = lhs_literal.data<ReturnT>();
1300 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs);
1305 return HandleConvolutionWithLiterals(conv, lhs_literal, rhs_literal);
1309 conv, lhs_literal.Convert(result_shape.element_type()).ValueOrDie(),
1314 conv, lhs_literal,
1318 conv, lhs_literal.Convert(result_shape.element_type()).ValueOrDie(),
1369 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs);
1375 lhs_array.SetValues(lhs_literal.data<NativeT>());
[all …]
|
| D | hlo_evaluator_test.cc | 801 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P() local
803 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
841 auto lhs_literal = LiteralUtil::CreateR1<float>({1, 2, 3}); in TEST_P() local
843 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
885 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); in TEST_P() local
887 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
928 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); in TEST_P() local
930 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
972 auto lhs_literal = LiteralUtil::CreateR3FromArray3D<float>(lhs_array); in TEST_P() local
974 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 415 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs); in ConvArray4DGeneralDimensionsDilated() local
430 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
432 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
467 lhs_literal.shape(), rhs_literal.shape(), in ConvArray4DGeneralDimensionsDilated()
473 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in ConvArray4DGeneralDimensionsDilated()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 261 Literal lhs_literal = LiteralUtil::CreateR1<uint64>({lhs}); in XLA_TEST_F() local
262 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param"); in XLA_TEST_F()
264 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
301 Literal lhs_literal = LiteralUtil::CreateR1<int64>({lhs}); in XLA_TEST_F() local
302 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param"); in XLA_TEST_F()
304 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
333 Literal lhs_literal = LiteralUtil::CreateR1<uint64>({lhs}); in XLA_TEST_F() local
334 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param"); in XLA_TEST_F()
342 ComputeAndCompare(&b, {std::move(lhs_literal), std::move(rhs_literal)}); in XLA_TEST_F()
|