HomeSort by relevance Sort by last modified time
    Searched refs:rhs_literal (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator_test.cc 629 auto rhs_literal = Literal::CreateR2<float>({{1, 2}}); local
631 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal)));
674 auto rhs_literal = Literal::CreateR2FromArray2D<float>(*rhs_array); local
676 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal)));
717 auto rhs_literal = Literal::CreateR2FromArray2D<float>(*rhs_array); local
719 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal)));
751 auto rhs_literal = Literal::CreateR3FromArray3D<float>(rhs_array); local
753 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal)));
813 auto rhs_literal = Literal::CreateR4FromArray4D<float>(rhs_array); local
815 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal)));
881 auto rhs_literal = Literal::CreateR4FromArray4D<float>(weight); local
961 auto rhs_literal = Literal::CreateR4FromArray4D<float>(weight); local
1032 auto rhs_literal = Literal::CreateR4FromArray4D<float>(rhs_array); local
1095 auto rhs_literal = Literal::CreateR4FromArray4D<float>(rhs_array); local
1160 auto rhs_literal = Literal::CreateR4FromArray4D<float>(rhs_array); local
    [all...]
hlo_evaluator.cc 63 const Literal& rhs_literal) {
105 rhs_literal.Get<OperandT>(multi_index));
114 const Literal& rhs_literal) {
136 rhs_literal.Get<complex64>(multi_index));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.cc 514 auto rhs_literal = Literal::CreateR4FromArray4D<float>(rhs); local
532 rhs_literal->shape().dimensions(dnums.kernel_spatial_dimensions(0));
534 rhs_literal->shape().dimensions(dnums.kernel_spatial_dimensions(1));
545 rhs_literal->shape().dimensions(dnums.kernel_spatial_dimensions(0)));
555 rhs_literal->shape().dimensions(dnums.kernel_spatial_dimensions(1)));
565 rhs_literal->shape(), window, dnums)
571 b.AddInstruction(HloInstruction::CreateConstant(std::move(rhs_literal)));

Completed in 127 milliseconds