Searched refs:HloBatchNormInstruction (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 76 HloBatchNormInstruction::HloBatchNormInstruction( in HloBatchNormInstruction() function in xla::HloBatchNormInstruction
86 bool HloBatchNormInstruction::IdenticalSlowPath( in IdenticalSlowPath()
90 const auto& casted_other = static_cast<const HloBatchNormInstruction&>(other); in IdenticalSlowPath()
95 HloInstructionProto HloBatchNormInstruction::ToProto() const { in ToProto()
102 std::vector<string> HloBatchNormInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()
111 : HloBatchNormInstruction(HloOpcode::kBatchNormTraining, shape, operand, in HloBatchNormTrainingInstruction()
130 : HloBatchNormInstruction(HloOpcode::kBatchNormInference, shape, operand, in HloBatchNormInferenceInstruction()
151 : HloBatchNormInstruction(HloOpcode::kBatchNormGrad, shape, operand, scale, in HloBatchNormGradInstruction()
|
| D | hlo_instructions.h | 28 class HloBatchNormInstruction : public HloInstruction {
42 explicit HloBatchNormInstruction(HloOpcode opcode, const Shape& shape,
61 class HloBatchNormTrainingInstruction : public HloBatchNormInstruction {
76 class HloBatchNormInferenceInstruction : public HloBatchNormInstruction {
90 class HloBatchNormGradInstruction : public HloBatchNormInstruction {
|
| D | hlo_instruction.cc | 3895 return Cast<HloBatchNormInstruction>(this)->feature_index(); in feature_index()
3899 return Cast<HloBatchNormInstruction>(this)->epsilon(); in epsilon()
|