| /external/tensorflow/tensorflow/compiler/xla/ |
| reference_util.cc | 481 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs); local
496 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(0));
498 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(1));
533 lhs_literal.shape(), rhs_literal.shape(),
538 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal)));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator.cc | 60 LiteralSlice lhs_literal, LiteralSlice rhs_literal) {
98 return compare_op(lhs_literal.Get<OperandT>(multi_index),
108 LiteralSlice lhs_literal,
130 return compare_op(lhs_literal.Get<complex64>(multi_index),
140 LiteralSlice lhs_literal,
162 return compare_op(lhs_literal.Get<complex128>(multi_index),
681 const Literal& lhs_literal = GetEvaluatedLiteralFor(lhs); local
689 Compare<bool>(compare->shape(), direction, lhs_literal, rhs_literal));
694 lhs_literal, rhs_literal));
699 lhs_literal, rhs_literal))
[all...] |
| hlo_evaluator_test.cc | 724 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); local
726 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal)));
764 auto lhs_literal = LiteralUtil::CreateR1<float>({1, 2, 3}); local
766 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal)));
808 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array); local
810 b.AddInstruction(HloInstruction::CreateConstant(std::move(lhs_literal)));
851 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*lhs_array); local
895 auto lhs_literal = LiteralUtil::CreateR3FromArray3D<float>(lhs_array); local
952 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs_array); local
1028 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(input); local
1109 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(input); local
1174 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs_array); local
1238 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs_array); local
1304 auto lhs_literal = LiteralUtil::CreateR4FromArray4D<float>(lhs_array); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| array_elementwise_ops_test.cc | 228 Literal lhs_literal = LiteralUtil::CreateR1<uint64>({lhs}); local
229 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param");
231 client_->TransferToServer(lhs_literal).ConsumeValueOrDie();
268 Literal lhs_literal = LiteralUtil::CreateR1<int64>({lhs}); local
269 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param");
271 client_->TransferToServer(lhs_literal).ConsumeValueOrDie();
300 Literal lhs_literal = LiteralUtil::CreateR1<uint64>({lhs}); local
301 auto lhs_param = Parameter(&b, 0, lhs_literal.shape(), "lhs_param");
309 ComputeAndCompare(&b, {std::move(lhs_literal), std::move(rhs_literal)});
[all...] |
