| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| prefetching_ops.py | 28 shared_name,
37 shared_name=shared_name,
|
| /external/tensorflow/tensorflow/python/training/ |
| input.py | 117 shared_name=None,
141 shared_name: (Optional.) If set, this queue will be shared under the given
182 shared_name=shared_name, name=name)
199 shared_name=None,
218 shared_name: (optional). If set, this queue will be shared under the given
220 this queue will be able to access it via the shared_name. Using this in
257 shared_name=shared_name,
265 capacity=32, shared_name=None, name=None)
[all...] |
| sync_replicas_optimizer.py | 273 shared_name=var.name + "/grad_accum")
282 grad.dtype, shape=(), shared_name=var.name + "/grad_accum")
304 shared_name="sync_token_q"))
315 shared_name="dummy_queue"))
|
| /external/tensorflow/tensorflow/contrib/nccl/python/ops/ |
| nccl_ops.py | 69 shared_name = op.get_attr('shared_name') + '_grad'
76 shared_name=shared_name)
217 shared_name = _get_shared_name()
228 shared_name=shared_name))
|
| /external/tensorflow/tensorflow/core/framework/ |
| resource_op_kernel_test.cc | 82 .Attr("shared_name: string = ''")
91 const string& shared_name) {
96 .Attr("shared_name", shared_name)
133 // Empty shared_name means private resource.
158 const string shared_name = "shared_stub"; local
160 auto op = CreateOp(code, shared_name);
165 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name,
173 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name,
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| prefetch_queue.py | 37 shared_name=None,
60 shared_name: (optional). If set, this queue will be shared under the given
85 shared_name=shared_name)
|
| /external/tensorflow/tensorflow/contrib/batching/python/ops/ |
| batch_ops.py | 40 shared_name="batch_gradient_{}_{}".format(op.name, i)))
52 shared_name="unbatch_gradient_{}".format(op.name)), None, None
121 shared_name=name)
131 shared_name=unbatch_name + "/" + t.name)
|
| batch_ops_test.py | 175 timeout_micros=1000000, shared_name="unbatch")
219 shared_name="shared_unbatch")
229 shared_name="shared_unbatch")
260 timeout_micros=1000000, shared_name="unbatch")
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| bucket_ops.py | 74 shared_name=None,
144 shared_name: (Optional). If set, the queues will be shared under the given
200 shared_name_i = ("%s_%d" % (shared_name, i) if shared_name is not None
207 shared_name=shared_name_i,
226 shared_name=shared_name,
314 shared_name=None,
353 shared_name: (Optional). If set, the queues will be shared under the given
420 shared_name=shared_name
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| data_flow_ops.py | 629 shared_name=None,
669 shared_name: (Optional.) If non-empty, this queue will be shared under
679 elif seed is None and shared_name is not None:
681 # If shared_name is also provided, make seed2 depend only on the graph
682 # seed and shared_name. (seed2 from get_seed() is generally dependent on
684 string = (str(seed1) + shared_name).encode("utf-8")
693 shared_name=shared_name,
717 shared_name=None,
744 shared_name: (Optional.) If non-empty, this queue will be shared unde
[all...] |
| state_ops.py | 98 shared_name=""):
103 container=container, shared_name=shared_name)
111 def variable_op_v2(shape, dtype, name="Variable", container="", shared_name=""):
123 shared_name: An optional string. Defaults to "".
125 with this shared_name. Otherwise, the node name is used instead.
134 shared_name=shared_name)
|
| lookup_ops.py | 256 def __init__(self, initializer, default_value, shared_name=None, name=None):
268 shared_name: If non-empty, this table will be shared under
279 shared_name=shared_name,
[all...] |
| resource_variable_ops.py | 42 def _eager_safe_variable_handle(shape, dtype, shared_name, name, graph_mode):
48 # When in eager mode use a uid for the shared_name, to prevent accidental
50 shared_name = str(ops.uid())
52 shared_name=shared_name,
66 shared_name)
69 shared_name=shared_name,
385 shared_name=handle_name,
400 shared_name=handle_name
[all...] |
| sparse_ops.py | [all...] |
| /external/tensorflow/tensorflow/python/data/ops/ |
| iterator_ops.py | 89 shared_name=None,
142 shared_name: (Optional.) If non-empty, this iterator will be shared under
166 if shared_name is None:
167 shared_name = ""
170 shared_name=shared_name,
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_rewrite.cc | 35 string shared_name = node->name(); local
37 return NodeBuilder(strings::StrCat(shared_name, suffix), op_name)
40 .Attr("shared_name", shared_name)
159 string shared_name = node->name(); local
161 return NodeBuilder(strings::StrCat(shared_name, suffix), op_name)
163 .Attr("shared_name", shared_name)
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| lookup_ops.py | 311 shared_name=None,
323 shared_name: If non-empty, this table will be shared under
327 from checkpoints. If `shared_name` is empty for a checkpointed table, it
341 # training to work correctly. Use the node name if no shared_name has been
343 use_node_name_sharing = checkpoint and shared_name is None
347 shared_name=shared_name,
354 shared_name=shared_name,
517 shared_name=None
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| random_shuffle_queue_test.py | [all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| barrier_ops_test.py | 39 shared_name="B",
66 attr { key: 'shared_name' value: { s: 'B' } }
682 (dtypes.float32,), shapes=(()), shared_name="b_a")
684 (dtypes.int32,), shapes=(()), shared_name="b_a")
690 (dtypes.float32,), shapes=(()), shared_name="b_b")
692 (dtypes.float32, dtypes.int32), shapes=((), ()), shared_name="b_b")
700 shared_name="b_c")
702 (dtypes.float32, dtypes.float32), shared_name="b_c")
708 (dtypes.float32, dtypes.float32), shapes=((), ()), shared_name="b_d")
712 shared_name="b_d"
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| graph_util_test.py | 55 shared_name="")
62 shared_name="")
68 shared_name="")
74 shared_name="")
81 shared_name="")
88 shared_name="")
94 shared_name="")
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| model_ops.py | 102 container, shared_name=name, name=name)
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
| model_ops.py | 106 container, shared_name=name, name=name)
|
| stats_ops.py | 102 container, shared_name=name, name=name)
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| datasets.py | 87 shared_name="",
122 shared_name=_generate_shared_name("function_buffer_resource"))
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| iterator_ops_cluster_test.py | 124 shared_name="shared_table")
132 .make_initializable_iterator(shared_name="shared_iterator"))
|
