| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 130 shared_name=None, argument
195 shared_name=shared_name, name=name)
217 shared_name=None, argument
275 shared_name=shared_name,
287 capacity=32, shared_name=None, name=None): argument
320 shared_name, "fraction_of_%d_full" % capacity, name)
330 capacity=32, shared_name=None, name=None): argument
374 shared_name=shared_name)
532 lambda: _store_sparse(t, shared_name=map_op_name),
538 lambda: _store_many_sparse(t, shared_name=map_op_name),
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | parameter_server.py | 36 def _eager_safe_variable_handle(shape, dtype, shared_name, name, graph_mode): argument
42 shared_name=shared_name,
50 shared_name=shared_name,
148 shared_name = handle_name
160 shared_name=shared_name,
172 shared_name=shared_name,
195 shared_name=shared_name,
200 self._unique_id = shared_name
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 613 def _shared_name(shared_name): argument
616 return shared_name
639 shared_name=None, argument
689 elif seed is None and shared_name is not None:
694 string = (str(seed1) + shared_name).encode("utf-8")
703 shared_name=_shared_name(shared_name),
723 shared_name=None, argument
761 shared_name=_shared_name(shared_name),
786 shared_name=None, argument
838 shared_name=_shared_name(shared_name),
[all …]
|
| D | nccl_ops.py | 69 shared_name = op.get_attr('shared_name') + b'_grad'
76 shared_name=shared_name)
215 shared_name = _get_shared_name()
228 shared_name=shared_name))
|
| D | critical_section_ops.py | 143 def __init__(self, name=None, shared_name=None, argument
152 self._init_from_args(name, shared_name)
166 def _init_from_args(self, name, shared_name): # pylint: disable=invalid-name argument
173 if shared_name is None:
174 shared_name = name
178 shared_name=shared_name, container=container, name=name)
|
| D | state_ops.py | 42 shared_name=""): argument
47 container=container, shared_name=shared_name)
55 def variable_op_v2(shape, dtype, name="Variable", container="", shared_name=""): argument
79 shared_name=shared_name)
|
| D | summary_ops_v2.py | 207 def __init__(self, shared_name, init_op_fn, name=None, v2=False): argument
209 shared_name=shared_name, name=name)
380 shared_name = context.shared_name()
382 shared_name = ops._name_from_scope_name(scope) # pylint: disable=protected-access
384 shared_name=shared_name,
433 shared_name=name,
485 shared_name=name,
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | indexed_dataset_ops.py | 76 def materialize(self, shared_name=None, container=None): argument
91 if shared_name is None:
92 shared_name = ""
96 shared_name=shared_name,
|
| D | threadpool.py | 48 shared_name = _generate_shared_name("privatethreadpool")
53 shared_name=shared_name)
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | bucket_ops.py | 74 shared_name=None, argument
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, argument
420 shared_name=shared_name)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_NcclAllReduce.pbtxt | 6 within the same `shared_name.
8 The graph should be constructed so if one op runs with shared_name value `c`,
9 then `num_devices` ops will run with shared_name value `c`. Failure to do so
16 shared_name: Identifier that shared between ops of the same reduction.
|
| D | api_def_AddSparseToTensorsMap.pbtxt | 35 name: "shared_name"
53 `container` and `shared_name` are passed to that Op. If no `shared_name`
55 `AddSparseToTensorsMap` as the `shared_name` passed to
|
| D | api_def_AddManySparseToTensorsMap.pbtxt | 37 name: "shared_name"
63 `container` and `shared_name` are passed to that Op. If no `shared_name`
65 `AddManySparseToTensorsMap` as the `shared_name` passed to
|
| D | api_def_Unbatch.pbtxt | 7 running instance of Unbatch with the same container and shared_name, or receives
20 shared_name: Instances of Unbatch with the same container and shared_name are
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel_test.cc | 91 const string& shared_name) { in CreateOp() argument
96 .Attr("shared_name", shared_name) in CreateOp()
158 const string shared_name = "shared_stub"; in TEST_F() local
160 auto op = CreateOp(code, shared_name); in TEST_F()
165 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
173 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/batching/python/ops/ |
| D | batch_ops.py | 42 shared_name="batch_gradient_{}_{}".format(op.name, i)))
54 shared_name="unbatch_gradient_{}".format(op.name)), None, None
126 shared_name=name,
184 shared_name=name)
194 shared_name=unbatch_name + "/" + t.name)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 113 shared_name=None, argument
192 if shared_name is None:
193 shared_name = ""
200 shared_name=shared_name,
206 shared_name=shared_name,
212 shared_name=shared_name,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | critical_section_test.py | 41 cs = critical_section_ops.CriticalSection(shared_name="cs")
62 cs = critical_section_ops.CriticalSection(shared_name="cs")
101 cs = critical_section_ops.CriticalSection(shared_name="cs")
128 cs = critical_section_ops.CriticalSection(shared_name="cs")
148 cs = critical_section_ops.CriticalSection(shared_name="cs")
194 cs = critical_section_ops.CriticalSection(shared_name="cs")
222 cs = critical_section_ops.CriticalSection(shared_name="cs")
310 cs = critical_section_ops.CriticalSection(shared_name="cs")
311 cs_same = critical_section_ops.CriticalSection(shared_name="cs")
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| D | prefetch_queue.py | 38 shared_name=None, argument
86 shared_name=shared_name)
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 37 string shared_name = node->name(); in ReplaceReduce() local
39 return NodeBuilder(strings::StrCat(shared_name, suffix), op_name) in ReplaceReduce()
42 .Attr("shared_name", shared_name) in ReplaceReduce()
161 string shared_name = node->name(); in ReplaceBroadcast() local
163 return NodeBuilder(strings::StrCat(shared_name, suffix), op_name) in ReplaceBroadcast()
165 .Attr("shared_name", shared_name) in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
| D | stats_ops.py | 109 shared_name = "fertile_stats_variable_%d" % (ops.uid(),)
111 shared_name = self._name
113 self._container, shared_name=shared_name, name=self._name)
|
| D | model_ops.py | 114 shared_name = "tree_variable_%d" % (ops.uid(),)
116 shared_name = self._name
118 self._container, shared_name=shared_name, name=self._name)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 273 …_many\', \'shapes\', \'dynamic_pad\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
277 …_many\', \'shapes\', \'dynamic_pad\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
341 …lement_shape\', \'num_epochs\', \'shuffle\', \'seed\', \'capacity\', \'shared_name\', \'summary_na…
377 …_many\', \'shapes\', \'dynamic_pad\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
381 …_many\', \'shapes\', \'dynamic_pad\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
385 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
389 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
413 …argspec: "args=[\'limit\', \'num_epochs\', \'shuffle\', \'seed\', \'capacity\', \'shared_name\', \…
437 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
441 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | random_shuffle_queue_test.py | 1074 1, 0, dtypes_lib.float32, ((),), shared_name="shared_queue")
1084 shared_name="shared_queue",
1114 shared_name="shared_queue",
1123 1, 0, dtypes_lib.float32, ((),), shared_name="shared_queue")
1134 10, 5, dtypes_lib.float32, shared_name="q_a")
1136 15, 5, dtypes_lib.float32, shared_name="q_a")
1142 10, 0, dtypes_lib.float32, shared_name="q_b")
1144 10, 5, dtypes_lib.float32, shared_name="q_b")
1150 10, 5, dtypes_lib.float32, shared_name="q_c")
1152 10, 5, dtypes_lib.int32, shared_name="q_c")
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_util_test.py | 57 shared_name="")
64 shared_name="")
70 shared_name="")
76 shared_name="")
83 shared_name="")
90 shared_name="")
96 shared_name="")
|