| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 129 shared_name=None, argument
194 shared_name=shared_name, name=name)
216 shared_name=None, argument
274 shared_name=shared_name,
286 capacity=32, shared_name=None, name=None): argument
319 shared_name, "fraction_of_%d_full" % capacity, name)
329 capacity=32, shared_name=None, name=None): argument
373 shared_name=shared_name)
531 lambda: _store_sparse(t, shared_name=map_op_name),
537 lambda: _store_many_sparse(t, shared_name=map_op_name),
[all …]
|
| /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
762 shared_name=_shared_name(shared_name),
787 shared_name=None, argument
826 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 | 195 def __init__(self, name=None, shared_name=None, argument
205 self._init_from_args(name, shared_name)
207 def _init_from_args(self, name, shared_name): # pylint: disable=invalid-name argument
214 if shared_name is None:
215 shared_name = name
219 shared_name=shared_name, container=container, name=name)
224 shared_name or id(self._handle),
|
| 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)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | decompose_resource_ops.mlir | 8 …%0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource<tensor<…
31 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
52 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
73 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
99 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
100 %1 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
128 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
129 %1 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
187 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
188 %1 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
[all …]
|
| D | promote_resources_to_args.mlir | 12 …%1 = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf.resource<tensor<f32…
32 …%1 = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf.resource<tensor<f32…
35 …%4 = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf.resource<tensor<f32…
55 …%1 = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf.resource<tensor<f32…
58 …%4 = "tf.VarHandleOp"() {container = "", shared_name = "y"} : () -> tensor<!tf.resource<tensor<f32…
77 …%1 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
106 …%1 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
|
| D | resource_op_lifting.mlir | 9 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
35 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
62 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
94 %1 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
120 %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource>
|
| /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
97 .Attr("shared_name", shared_name) in CreateOp()
161 const string shared_name = "shared_stub"; in TEST_F() local
163 auto op = CreateOp(code, shared_name); in TEST_F()
168 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
176 TF_ASSERT_OK(mgr_.Lookup<StubResource>(mgr_.default_container(), shared_name, in TEST_F()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | critical_section_test.py | 47 cs = critical_section_ops.CriticalSection(shared_name="cs")
73 cs = critical_section_ops.CriticalSection(shared_name="cs")
112 cs = critical_section_ops.CriticalSection(shared_name="cs")
139 cs = critical_section_ops.CriticalSection(shared_name="cs")
159 cs = critical_section_ops.CriticalSection(shared_name="cs")
202 cs = critical_section_ops.CriticalSection(shared_name="cs")
230 cs = critical_section_ops.CriticalSection(shared_name="cs")
317 cs = critical_section_ops.CriticalSection(shared_name="cs")
318 cs_same = critical_section_ops.CriticalSection(shared_name="cs")
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | promote_resources_to_args.cc | 92 if (resource_map.count(var_handle_op.shared_name())) { in PromoteResourcesToArguments()
104 resource_map[var_handle_op.shared_name()] = {input_num++, arg}; in PromoteResourcesToArguments()
124 resource_map[var_handle_op.shared_name()].live_value); in PromoteResourcesToArguments()
132 resource_map[var_handle_op.shared_name()].live_value = write_op.value(); in PromoteResourcesToArguments()
152 ResourceInfo& resource_info = resource_map[var_handle_op.shared_name()]; in PromoteResourcesToArguments()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseConditionalAccumulator.pbtxt | 42 name: "shared_name"
93 name: "shared_name"
145 name: "shared_name"
197 name: "shared_name"
249 name: "shared_name"
|
| D | ConditionalAccumulator.pbtxt | 42 name: "shared_name"
93 name: "shared_name"
145 name: "shared_name"
197 name: "shared_name"
249 name: "shared_name"
|
| D | FixedLengthRecordReaderV2.pbtxt | 33 name: "shared_name"
80 name: "shared_name"
127 name: "shared_name"
|
| D | FixedLengthRecordReader.pbtxt | 34 name: "shared_name"
82 name: "shared_name"
130 name: "shared_name"
|
| D | IdentityReader.pbtxt | 16 name: "shared_name"
39 name: "shared_name"
|
| D | TFRecordReader.pbtxt | 16 name: "shared_name"
46 name: "shared_name"
|
| D | TextLineReader.pbtxt | 23 name: "shared_name"
53 name: "shared_name"
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | threadpool.py | 48 shared_name = _generate_shared_name("privatethreadpool")
53 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/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…
345 …lement_shape\', \'num_epochs\', \'shuffle\', \'seed\', \'capacity\', \'shared_name\', \'summary_na…
381 …_many\', \'shapes\', \'dynamic_pad\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
385 …_many\', \'shapes\', \'dynamic_pad\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
389 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
393 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
417 …argspec: "args=[\'limit\', \'num_epochs\', \'shuffle\', \'seed\', \'capacity\', \'shared_name\', \…
441 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
445 …seed\', \'enqueue_many\', \'shapes\', \'allow_smaller_final_batch\', \'shared_name\', \'name\'], v…
[all …]
|