Searched refs:operand_literal (Results 1 – 2 of 2) sorted by relevance
146 const Literal& operand_literal) { in ElementWiseUnaryOpImpl() argument164 return unary_op(operand_literal.Get<NativeT>(multi_index)); in ElementWiseUnaryOpImpl()871 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand); in HandleReverse() local880 return operand_literal.Get<ReturnT>(from_index); in HandleReverse()1244 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand); in HandleDynamicSlice() local1252 DynamicSlice<int32>(operand_literal, start_indices_literal, in HandleDynamicSlice()1258 DynamicSlice<int64>(operand_literal, start_indices_literal, in HandleDynamicSlice()1264 DynamicSlice<uint32>(operand_literal, start_indices_literal, in HandleDynamicSlice()1270 DynamicSlice<uint64>(operand_literal, start_indices_literal, in HandleDynamicSlice()1300 const Literal& operand_literal = parent_->GetEvaluatedLiteralFor(operand); in HandleDynamicUpdateSlice() local[all …]
1432 auto operand_literal = Literal::CreateR2FromArray2D<float>(*operand_array); in TEST_P() local1435 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()1464 auto operand_literal = Literal::CreateR2FromArray2D<float>(*operand_array); in TEST_P() local1467 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()1499 auto operand_literal = Literal::CreateR2FromArray2D<float>(*operand_array); in TEST_P() local1502 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()1532 auto operand_literal = Literal::CreateR2FromArray2D<double>(*operand_array); in TEST_P() local1535 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()1654 auto operand_literal = Literal::CreateR4FromArray4D<float>(input); in TEST_P() local1656 HloInstruction::CreateConstant(std::move(operand_literal))); in TEST_P()