Searched refs:HloPadInstruction (Results 1 – 4 of 4) sorted by relevance
2196 HloPadInstruction::HloPadInstruction(const Shape& shape, in HloPadInstruction() function in xla::HloPadInstruction2205 HloInstructionProto HloPadInstruction::ToProto() const { in ToProto()2211 std::vector<string> HloPadInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()2216 bool HloPadInstruction::IdenticalSlowPath( in IdenticalSlowPath()2220 const auto& casted_other = static_cast<const HloPadInstruction&>(other); in IdenticalSlowPath()2225 std::unique_ptr<HloInstruction> HloPadInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()2229 return absl::make_unique<HloPadInstruction>(shape, new_operands[0], in CloneWithNewOperandsImpl()
1253 class HloPadInstruction : public HloInstruction {1255 explicit HloPadInstruction(const Shape& shape, HloInstruction* operand,
1170 return absl::make_unique<HloPadInstruction>(shape, operand, padding_value, in CreatePad()3505 return Cast<HloPadInstruction>(this)->padding_config(); in padding_config()
2682 auto pad = Cast<HloPadInstruction>(slice->mutable_operand(0)); in TrySimplifyScalarSlice()