Home
last modified time | relevance | path

Searched refs:grad_string (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dexport_graphdef.cc624 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction() local
626 function->getAttrOfType<mlir::FlatSymbolRefAttr>(grad_string)) { in ConvertLibFunction()
645 grad_string.data(), stateful_string.data(), kEntryFuncAttr}; in ConvertLibFunction()
H A Dimport_model.cc1429 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction() local
1430 attributes.push_back(builder_.getNamedAttr(grad_string, gradient_attr)); in ConvertLibFunction()