Searched refs:resource_handles (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | batch_ops_utils.py | 88 resource_handles = []
95 resource_handles.append(op.resource_handle)
98 (k, _move_tensors(v, resource_handles[0].device))
100 with ops.device(resource_handles[0].device):
101 return batch[0].op(resource_handles, stamp, **grouped_args)
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.h | 143 resource_handles,
162 resource_handles,
169 resource_handles,
|
| D | graph_properties.cc | 1865 const std::unordered_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateShapes() argument
1878 UpdateEnqueue(n, resource_handles, shape_refiner, new_shapes)); in UpdateShapes()
1894 const std::unordered_map<const NodeDef*, const NodeDef*>& resource_handles, in PropagateShapes() argument
1901 << num_loops << " loops and " << resource_handles.size() in PropagateShapes()
1908 const int64 num_queues = resource_handles.size(); in PropagateShapes()
1919 UpdateShapes(shape_refiner, resource_handles, n, &updated)); in PropagateShapes()
1927 auto it = resource_handles.find(n); in PropagateShapes()
1928 if (it != resource_handles.end()) { in PropagateShapes()
1993 const std::unordered_map<const NodeDef*, const NodeDef*>& resource_handles, in UpdateEnqueue() argument
2001 auto it = resource_handles.find(enqueue_node); in UpdateEnqueue()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 83 std::vector<const ResourceHandle*> resource_handles; in GetVariableInfosFromCtxInputs() local
85 variable_indices, std::back_inserter(resource_handles), in GetVariableInfosFromCtxInputs()
89 TF_RETURN_IF_ERROR(LookupResources(ctx, resource_handles, &variables)); in GetVariableInfosFromCtxInputs()
|