Searched refs:HloPadInstruction (Results 1 – 3 of 3) sorted by relevance
2932 HloPadInstruction::HloPadInstruction(const Shape& shape, in HloPadInstruction() function in xla::HloPadInstruction2941 HloInstructionProto HloPadInstruction::ToProto() const { in ToProto()2947 std::vector<std::string> HloPadInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()2952 bool HloPadInstruction::IdenticalSlowPath( in IdenticalSlowPath()2956 const auto& casted_other = static_cast<const HloPadInstruction&>(other); in IdenticalSlowPath()2961 std::unique_ptr<HloInstruction> HloPadInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()2965 return std::make_unique<HloPadInstruction>(shape, new_operands[0], in CloneWithNewOperandsImpl()
1973 class HloPadInstruction : public HloInstruction {1975 explicit HloPadInstruction(const Shape& shape, HloInstruction* operand,
1741 return std::make_unique<HloPadInstruction>(shape, operand, padding_value, in CreatePad()4793 return Cast<HloPadInstruction>(this)->padding_config(); in padding_config()4801 return Cast<HloPadInstruction>(this)->mutable_padding_config(); in mutable_padding_config()