Searched refs:HloConstantInstruction (Results 1 – 8 of 8) sorted by relevance
38 HloConstantInstruction* alpha_conv;39 HloConstantInstruction* alpha_side_input;168 CastOrNull<HloConstantInstruction>(alpha_conv_instr), in FindConvWithRelu()169 CastOrNull<HloConstantInstruction>(alpha_side_input_instr)}; in FindConvWithRelu()180 [](HloConstantInstruction* instr) -> StatusOr<double> { in TryRewriteToCudnnForwardRelu()183 Cast<HloConstantInstruction>(instr)->literal().Convert(F64)); in TryRewriteToCudnnForwardRelu()
1024 HloConstantInstruction::HloConstantInstruction(Literal literal) in HloConstantInstruction() function in xla::HloConstantInstruction1028 HloConstantInstruction::HloConstantInstruction(const Shape& shape) in HloConstantInstruction() function in xla::HloConstantInstruction1031 HloInstructionProto HloConstantInstruction::ToProto() const { in ToProto()1039 bool HloConstantInstruction::IsElementwiseImpl( in IsElementwiseImpl()1044 void HloConstantInstruction::RelayoutConstant(const Layout& new_layout, in RelayoutConstant()1060 bool HloConstantInstruction::IdenticalSlowPath( in IdenticalSlowPath()1069 HloConstantInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1073 return absl::make_unique<HloConstantInstruction>(literal_->Clone()); in CloneWithNewOperandsImpl()1076 string HloConstantInstruction::OperandsToStringWithCanonicalNameMap( in OperandsToStringWithCanonicalNameMap()
717 static const HloConstantInstruction* TryGetFusionParameterConstant( in TryGetFusionParameterConstant()724 return DynCast<HloConstantInstruction>(operand); in TryGetFusionParameterConstant()816 auto stringify_constant = [](const HloConstantInstruction* constant) { in GetInstructionNodeInlinedOperands()856 const auto* constant_operand = DynCast<HloConstantInstruction>(operand); in GetInstructionNodeInlinedOperands()867 if (const HloConstantInstruction* constant = in GetInstructionNodeInlinedOperands()
650 class HloConstantInstruction : public HloInstruction {652 explicit HloConstantInstruction(Literal literal);654 explicit HloConstantInstruction(const Shape& shape);
790 absl::flat_hash_map<int64, const HloConstantInstruction*> induction_vars; in TryMergeInductionVariables()806 induction_vars.emplace(i, Cast<HloConstantInstruction>(constant)); in TryMergeInductionVariables()
307 instruction = absl::make_unique<HloConstantInstruction>(shape); in CreateFromProto()650 return absl::make_unique<HloConstantInstruction>(std::move(literal)); in CreateConstant()3272 return Cast<HloConstantInstruction>(this)->literal(); in literal()3276 return DynCast<HloConstantInstruction>(this) != nullptr; in IsConstant()3281 Cast<HloConstantInstruction>(this)->RelayoutConstant(new_layout, shape_index); in RelayoutConstant()
392 if (!Cast<HloConstantInstruction>(constant)->HasLiteral()) { in HandleConstant()
1738 const auto* const_inst = DynCast<HloConstantInstruction>(inst);