Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc623 auto lhs_literal = Literal::CreateR2FromArray2D<float>(*lhs_array); in TEST_P() local
625 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
662 auto lhs_literal = Literal::CreateR1<float>({1, 2, 3}); in TEST_P() local
664 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
705 auto lhs_literal = Literal::CreateR2FromArray2D<float>(*lhs_array); in TEST_P() local
707 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
746 auto lhs_literal = Literal::CreateR3FromArray3D<float>(lhs_array); in TEST_P() local
748 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
802 auto lhs_literal = Literal::CreateR4FromArray4D<float>(lhs_array); in TEST_P() local
804 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in TEST_P()
[all …]
Dhlo_evaluator.cc62 const Literal& lhs_literal, in Compare() argument
104 return compare_op(lhs_literal.Get<OperandT>(multi_index), in Compare()
113 const Shape& shape, HloOpcode opcode, const Literal& lhs_literal, in Compare() argument
135 return compare_op(lhs_literal.Get<complex64>(multi_index), in Compare()
923 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in HandleConvolution() local
1010 static_cast<ElementwiseT>(lhs_literal.Get<ReturnT>(lhs_index)) * in HandleConvolution()
1056 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in HandleDot() local
1148 static_cast<ElementwiseT>(lhs_literal.Get<ReturnT>(lhs_index)) * in HandleDot()
1981 const Literal& lhs_literal = parent_->GetEvaluatedLiteralFor(lhs); in ElementWiseBinaryOp() local
1989 lhs_literal.Get<ReturnT>(multi_index), in ElementWiseBinaryOp()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc513 auto lhs_literal = Literal::CreateR4FromArray4D<float>(lhs); in ConvArray4DGeneralDimensionsDilated() local
528 lhs_literal->shape().dimensions(dnums.input_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
530 lhs_literal->shape().dimensions(dnums.input_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
564 ShapeInference::InferConvolveShape(lhs_literal->shape(), in ConvArray4DGeneralDimensionsDilated()
569 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal))); in ConvArray4DGeneralDimensionsDilated()