Searched defs:rhs_literal (Results 1 – 5 of 5) sorted by relevance
730 auto rhs_literal = LiteralUtil::CreateR2<float>({{1, 2}}); in TEST_P() local776 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P() local820 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array); in TEST_P() local857 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(*rhs_array); in TEST_P() local900 auto rhs_literal = LiteralUtil::CreateR3FromArray3D<float>(rhs_array); in TEST_P() local963 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs_array); in TEST_P() local1032 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(weight); in TEST_P() local1113 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(weight); in TEST_P() local1185 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs_array); in TEST_P() local1249 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs_array); in TEST_P() local[all …]
1047 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in HandleConvolution() local1250 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in HandleDot() local1291 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in HandleDotSlowPath() local2954 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in ElementWiseBinaryOp() local2980 const Literal& rhs_literal = parent_->GetEvaluatedLiteralFor(rhs); in ElementwiseTernaryOp() local
60 LiteralSlice lhs_literal, LiteralSlice rhs_literal) { in Compare()109 LiteralSlice rhs_literal) { in Compare()141 LiteralSlice rhs_literal) { in Compare()682 const Literal& rhs_literal = GetEvaluatedLiteralFor(rhs); in HandleCompare() local
482 auto rhs_literal = LiteralUtil::CreateR4FromArray4D<float>(rhs); in ConvArray4DGeneralDimensionsDilated() local
242 Literal rhs_literal = LiteralUtil::CreateR1<uint64>({rhs}); in XLA_TEST_F() local281 Literal rhs_literal = LiteralUtil::CreateR1<int64>({rhs}); in XLA_TEST_F() local304 Literal rhs_literal = LiteralUtil::CreateR1<uint64>({rhs}); in XLA_TEST_F() local