| /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/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | force_shared_name_for_resource_ops.pbtxt | 13 key: "shared_name"
42 key: "shared_name"
65 key: "shared_name"
117 key: "shared_name"
137 # CHECK-SAME: shared_name = "hash_table_node"
140 # CHECK-SAME: shared_name = "variable_node"
142 # CHECK-SAME: shared_name = "variable_v2_node"
146 # CHECK-SAME: shared_name = "hash_table_node@create_resource"
|
| /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),
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | promote_var_handles_to_args.mlir | 18 …%0 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
26 …%0 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
34 …%0 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
35 …%1 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "y"} : () -> tensor<!tf…
43 …%0 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
44 …%1 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
54 …%0 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
56 …%2 = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf…
|
| D | readonly_references_to_resources.mlir | 8 …lass = ["loc:@v"], container = "", device = "", shape = #tf.shape<96>, shared_name = ""} : () -> t…
20 …%val0 = "tf.VariableV2"() {container = "", device = "", shape = #tf.shape<96>, shared_name = ""} :…
38 …lass = ["loc:@v"], container = "", device = "", shape = #tf.shape<96>, shared_name = ""} : () -> t…
51 …lass = ["loc:@v"], container = "", device = "", shape = #tf.shape<96>, shared_name = ""} : () -> t…
61 …lass = ["loc:@v"], container = "", device = "", shape = #tf.shape<96>, shared_name = ""} : () -> t…
72 …%val0 = "tf.VariableV2"() {container = "", device = "", shape = #tf.shape<96>, shared_name = ""} :…
83 …unrelated_class"], container = "", device = "", shape = #tf.shape<96>, shared_name = ""} : () -> t…
94 …:@v1", "loc:@v2"], container = "", device = "", shape = #tf.shape<96>, shared_name = ""} : () -> t…
|
| 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<!tf.resource<tensor<i3…
52 …%0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf.resource<tensor<i3…
73 …%0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf.resource<tensor<f3…
99 …%0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf.resource<tensor<f3…
100 …%1 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf.resource<tensor<f3…
125 …%0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf.resource<tensor<f3…
126 …%1 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf.resource<tensor<f3…
154 …%0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource<tensor<…
155 …%1 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf.resource<tensor<…
[all …]
|
| D | tf_saved_model_lift_variables_invalid_session.mlir | 9 …oc:@dense/kernel"], allowed_devices = [], container = "", device = "", shared_name = "dense/kernel…
11 …"loc:@dense/bias"], allowed_devices = [], container = "", device = "", shared_name = "dense/bias"}…
26 …oc:@dense/kernel"], allowed_devices = [], container = "", device = "", shared_name = "dense/kernel…
28 …"loc:@dense/bias"], allowed_devices = [], container = "", device = "", shared_name = "dense/bias"}…
|
| D | resource-alias-analysis-test.mlir | 20 %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
42 %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
44 %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
80 %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
119 %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
165 %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
203 %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
215 %vx = "tf.VarHandleOp"() {container = "cf", shared_name = "vx"} : () -> !tf_res
229 %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
258 %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
[all …]
|
| /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
|
| /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/python/data/experimental/ops/ |
| D | threadpool.py | 48 shared_name = _generate_shared_name("privatethreadpool")
53 shared_name=shared_name)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ConditionalAccumulator.pbtxt | 42 name: "shared_name"
93 name: "shared_name"
145 name: "shared_name"
197 name: "shared_name"
249 name: "shared_name"
|
| D | SparseConditionalAccumulator.pbtxt | 42 name: "shared_name"
93 name: "shared_name"
145 name: "shared_name"
197 name: "shared_name"
249 name: "shared_name"
|
| D | FixedLengthRecordReader.pbtxt | 34 name: "shared_name"
82 name: "shared_name"
130 name: "shared_name"
|
| D | FixedLengthRecordReaderV2.pbtxt | 33 name: "shared_name"
80 name: "shared_name"
127 name: "shared_name"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ConditionalAccumulator.pbtxt | 42 name: "shared_name"
93 name: "shared_name"
145 name: "shared_name"
197 name: "shared_name"
249 name: "shared_name"
|
| D | SparseConditionalAccumulator.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"
|
| /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()
|