Searched defs:gradient_type (Results 1 – 5 of 5) sorted by relevance
55 auto gradient_type = input_args[kInputIndex9]->GetType(); in LambInferType() local
83 DataType gradient_type = gradient_tensor->GetDataType(); in ParamCheck() local
407 const auto gradient_type = shader->asAGradient(&grInfo); in addGradientShaderResources() local
760 std::string gradient_type = (common::AnfAlgo::HasNodeAttr(kAttrGradientType, ps_optimizer)) in ProcessForSplitOptimizer() local843 … const std::string &gradient_type, in CreateNodesForGradAccumulation()
442 const auto gradient_type = as_SB(shader)->asGradient(&grInfo); in addGradientShaderResources() local