Searched refs:tensor_key (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 305 TensorKey tensor_key = ntohl(wc[i].imm_data); in Run() local
313 auto iter = tensor_buffers_.find(tensor_key); in Run()
316 << tensor_key; in Run()
391 TensorKey tensor_key = next_key_++; in TransportOptionsFromTensor() local
395 tensor_buffers_.insert(std::make_pair(tensor_key, buffer)); in TransportOptionsFromTensor()
403 remote_mr.set_tensor_key(tensor_key); in TransportOptionsFromTensor()
476 TensorKey tensor_key = remote_mr.tensor_key(); in TensorFromTransportOptions() local
479 tensor, start, tensor_key](const Status& s) { in TensorFromTransportOptions()
488 VLOG(2) << "RDMA of tensor " << tensor_key << " of size " in TensorFromTransportOptions()
511 if (tensor_callbacks_.find(tensor_key) == std::end(tensor_callbacks_)) { in TensorFromTransportOptions()
[all …]
|
| D | gdr.proto | 11 uint32 tensor_key = 5; field
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | grpc_debug_server.py | 260 tensor_key = "%s_%s_%d" % (device_name, debug_node_name, timestamp)
262 if tensor_key not in tensor_chunks:
263 tensor_chunks[tensor_key] = [None] * num_chunks
265 chunks = tensor_chunks[tensor_key]
275 del tensor_chunks[tensor_key]
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| D | tfexample_decoder.py | 152 def __init__(self, tensor_key, shape_keys=None, shape=None, default_value=0): argument
181 self._tensor_key = tensor_key
185 keys = [tensor_key]
215 tensor_key, argument
239 super(LookupTensor, self).__init__(tensor_key, shape_keys, shape,
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 328 outputs_tensor_info[tensor_key].name
329 for tensor_key in output_tensor_keys_sorted
|