Searched refs:HloBatchNormInstruction (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 73 HloBatchNormInstruction::HloBatchNormInstruction( in HloBatchNormInstruction() function in xla::HloBatchNormInstruction
83 bool HloBatchNormInstruction::IdenticalSlowPath( in IdenticalSlowPath()
87 const auto& casted_other = static_cast<const HloBatchNormInstruction&>(other); in IdenticalSlowPath()
92 HloInstructionProto HloBatchNormInstruction::ToProto() const { in ToProto()
99 std::vector<string> HloBatchNormInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()
108 : HloBatchNormInstruction(HloOpcode::kBatchNormTraining, shape, operand, in HloBatchNormTrainingInstruction()
127 : HloBatchNormInstruction(HloOpcode::kBatchNormInference, shape, operand, in HloBatchNormInferenceInstruction()
148 : HloBatchNormInstruction(HloOpcode::kBatchNormGrad, shape, operand, scale, in HloBatchNormGradInstruction()
|
| D | hlo_instructions.h | 26 class HloBatchNormInstruction : public HloInstruction {
40 explicit HloBatchNormInstruction(HloOpcode opcode, const Shape& shape,
59 class HloBatchNormTrainingInstruction : public HloBatchNormInstruction {
74 class HloBatchNormInferenceInstruction : public HloBatchNormInstruction {
88 class HloBatchNormGradInstruction : public HloBatchNormInstruction {
|
| D | hlo_instruction.cc | 3215 return Cast<HloBatchNormInstruction>(this)->feature_index(); in feature_index()
3219 return Cast<HloBatchNormInstruction>(this)->epsilon(); in epsilon()
|