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.cc159 auto resource_attr = resource_op->getAttr(kDeviceAttr); in BuildVariableAccessInfo() local
161 if (!resource_attr || resource_attr != device_attr) continue; in BuildVariableAccessInfo()
167 auto resource_attr = func.getArgAttrOfType<mlir::StringAttr>( in BuildVariableAccessInfo() local
169 if (!resource_attr || resource_attr != device_attr) continue; in BuildVariableAccessInfo()