Searched refs:grad_string (Results 1 – 2 of 2) sorted by relevance
752 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction() local754 function.getAttrOfType<mlir::FlatSymbolRefAttr>(grad_string)) { in ConvertLibFunction()782 grad_string.data(), stateful_string.data()}; in ConvertLibFunction()
1101 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction() local1102 attributes.push_back(builder_.getNamedAttr(grad_string, gradient_attr)); in ConvertLibFunction()