Searched refs:grad_string (Results 1 – 2 of 2) sorted by relevance
649 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction() local651 function->getAttrOfType<mlir::FlatSymbolRefAttr>(grad_string)) { in ConvertLibFunction()671 grad_string.data(), stateful_string.data()}; in ConvertLibFunction()
1281 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction() local1282 attributes.push_back(builder_.getNamedAttr(grad_string, gradient_attr)); in ConvertLibFunction()