OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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