Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc186 auto resource_attr = resource_op->getAttr(kDeviceAttr); in BuildVariableAccessInfo() local
188 if (!resource_attr || resource_attr != device_attr) continue; in BuildVariableAccessInfo()
194 auto resource_attr = func.getArgAttrOfType<mlir::StringAttr>( in BuildVariableAccessInfo() local
196 if (!resource_attr || resource_attr != device_attr) continue; in BuildVariableAccessInfo()