Home
last modified time | relevance | path

Searched refs:GetGradientAttrName (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.h61 static StringRef GetGradientAttrName() { return "tf.gradient"; } in GetGradientAttrName() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc649 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction()
Dimport_model.cc1281 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction()