Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtensorflow_abi_knowledge_propagation.cc60 allocated_by_tf_runtime.insert(argument); in runOnFunction()
74 allocated_by_tf_runtime.insert(allocated); in runOnFunction()
103 if (allocated_by_tf_runtime.contains(operand)) { in runOnFunction()
161 if (allocated_by_tf_runtime.contains(candidate)) { in propagateThroughUses()
162 allocated_by_tf_runtime.insert(result); in propagateThroughUses()
179 if (allocated_by_tf_runtime.contains(candidate)) { in propagateThroughUses()
180 allocated_by_tf_runtime.insert(result); in propagateThroughUses()
201 llvm::SmallPtrSet<Value, 8> allocated_by_tf_runtime; member