Searched refs:kInputIndex4 (Results 1 – 7 of 7) sorted by relevance
32 auto gamma_backprob = input_args[kInputIndex4]->Broaden(); in LayerNormGradInfer()33 auto beta_backprob = input_args[kInputIndex4]->Broaden(); in LayerNormGradInfer()
64 auto reserve_2 = input_args[kInputIndex4]->Broaden(); in BatchNormGradInfer()
41 …auto lr_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->BuildSha… in InferShape()71 auto lr_type = input_args[kInputIndex4]->BuildType(); in InferType()
78 auto m_type = input_args[kInputIndex4]->BuildType(); in ApplyMomentumInfer()
87 …auto variance = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->BuildSha… in BatchNormInfer()
95 auto infer_type4 = input_args[kInputIndex4]->BuildType()->cast<TensorTypePtr>()->element(); in LstmInfer()
266 kInputIndex4, enumerator