Searched refs:resource_handles (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.h | 173 resource_handles,
192 resource_handles,
199 resource_handles,
|
| D | graph_properties.cc | 2298 const absl::flat_hash_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateShapes() argument
2311 UpdateEnqueue(n, resource_handles, shape_refiner, new_shapes)); in UpdateShapes()
2327 const absl::flat_hash_map<const NodeDef*, const NodeDef*>& resource_handles, in PropagateShapes() argument
2334 << num_loops << " loops and " << resource_handles.size() in PropagateShapes()
2341 const int64 num_queues = resource_handles.size(); in PropagateShapes()
2352 UpdateShapes(shape_refiner, resource_handles, n, &updated)); in PropagateShapes()
2360 auto it = resource_handles.find(n); in PropagateShapes()
2361 if (it != resource_handles.end()) { in PropagateShapes()
2426 const absl::flat_hash_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateEnqueue() argument
2434 auto it = resource_handles.find(enqueue_node); in UpdateEnqueue()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2153 auto resource_handles = llvm::make_filter_range(args(), [](Value value) { in getEffects() local
2160 for (auto &entry : llvm::enumerate(resource_handles)) { in getEffects()
|