HomeSort by relevance Sort by last modified time
    Searched refs:tensor_key (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/gdr/
gdr_memory_manager.cc 305 TensorKey tensor_key = ntohl(wc[i].imm_data); local
313 auto iter = tensor_buffers_.find(tensor_key);
316 << tensor_key;
344 << ": ibv_post_send failed for tensor_key " << key;
391 TensorKey tensor_key = next_key_++; local
395 tensor_buffers_.insert(std::make_pair(tensor_key, buffer));
403 remote_mr.set_tensor_key(tensor_key);
476 TensorKey tensor_key = remote_mr.tensor_key(); local
479 tensor, start, tensor_key](const Status& s)
    [all...]
  /external/tensorflow/tensorflow/contrib/slim/python/slim/data/
tfexample_decoder.py 152 def __init__(self, tensor_key, shape_keys=None, shape=None, default_value=0):
157 is provided, both the `Tensor` corresponding to `tensor_key` and
160 corresponding to `tensor_key` is loaded and reshape appropriately.
165 tensor_key: the name of the `TFExample` feature to read the tensor from.
171 default_value: The value used when the `tensor_key` is not found in a
181 self._tensor_key = tensor_key
185 keys = [tensor_key]
215 tensor_key,
225 tensor_key: the name of the `TFExample` feature to read the tensor from.
232 default_value: The value used when the `tensor_key` is not found in
    [all...]
  /external/tensorflow/tensorflow/python/debug/lib/
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/python/tools/
saved_model_cli.py 328 outputs_tensor_info[tensor_key].name
329 for tensor_key in output_tensor_keys_sorted
    [all...]

Completed in 418 milliseconds