| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 34 output_arg {
57 output_arg {
120 output_arg {
148 output_arg {
181 output_arg {
211 output_arg {
242 output_arg {
274 output_arg {
306 output_arg {
339 output_arg {
[all …]
|
| D | ops_history.v1.pbtxt | 24 output_arg {
48 output_arg {
73 output_arg {
99 output_arg {
147 output_arg {
196 output_arg {
405 output_arg {
433 output_arg {
471 output_arg {
511 output_arg {
[all …]
|
| D | ops_history.v2.pbtxt | 24 output_arg {
48 output_arg {
73 output_arg {
99 output_arg {
147 output_arg {
196 output_arg {
405 output_arg {
433 output_arg {
471 output_arg {
511 output_arg {
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 24 output_arg {
50 output_arg {
140 output_arg {
168 output_arg {
204 output_arg {
231 output_arg {
260 output_arg {
299 output_arg {
330 output_arg {
383 output_arg {
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils_test.cc | 44 for (const auto &output_arg : function.signature().output_arg()) in CheckUniqueNames() local
45 outputs.insert(output_arg.name()); in CheckUniqueNames()
87 fused_function->signature().output_arg(0).name()); in TEST()
118 EXPECT_EQ(xtimes_two->signature().output_arg(0).name(), in TEST()
119 fused_function->signature().output_arg(0).name()); in TEST()
121 EXPECT_EQ(fused_function->signature().output_arg(1).name(), in TEST()
125 fused_function->signature().output_arg(0).name()); in TEST()
128 fused_function->signature().output_arg(1).name()); in TEST()
162 function_signature.output_arg()); in TEST()
|
| D | fusion_utils.cc | 52 function.signature().output_arg(output_idx).name(); in GetOutputNode()
58 function->signature().output_arg(output_idx).name(); in GetMutableOutputNode()
169 first_signature.output_arg(), second_signature.output_arg()); in GetUniqueSignature()
171 for (const auto& output_arg : second_signature.output_arg()) { in GetUniqueSignature() local
173 output = output_arg; in GetUniqueSignature()
327 *fused_signature->mutable_output_arg() = second_signature.output_arg(); in ComposeSignature()
344 second_signature.output_arg()); in CombineSignature()
|
| D | map_vectorization.cc | 104 const auto& output_arg = vectorized_func->signature().output_arg(i); in CreateMapDefunWrapper() local
105 (*vectorized_func->mutable_ret())[output_arg.name()] = in CreateMapDefunWrapper()
335 auto* output_arg = signature->add_output_arg(); in AddBranch() local
336 output_arg->set_name("output"); in AddBranch()
337 output_arg->set_type(DT_VARIANT); in AddBranch()
|
| D | vectorization_utils_test.cc | 66 for (const auto& output_arg : inner.signature().output_arg()) { in WrapFunctionWithMapDefun() local
67 output_types.push_back(output_arg.type()); in WrapFunctionWithMapDefun()
120 function_def.signature().output_arg(index).name()); in GetRetval()
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | update_api_def.cc | 51 for (auto& output_arg : op.output_arg()) { in FillBaseApiDef() local
52 if (!output_arg.description().empty()) { in FillBaseApiDef()
54 api_def_out_arg->set_name(output_arg.name()); in FillBaseApiDef()
55 api_def_out_arg->set_description(output_arg.description()); in FillBaseApiDef()
81 for (const auto& arg : op.output_arg()) { in OpHasDocs()
111 if (op1.output_arg(i).description() != op2.output_arg(i).description()) { in CheckDocsMatch()
134 for (const auto& arg : op.output_arg()) { in ValidateOpDocs()
|
| D | api_test.cc | 116 for (const auto& op_arg : op.output_arg()) { in TestAllApiDefOutputArgsAreValid()
252 for (const auto& arg : op.output_arg()) { in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 333 for (const OutputArgExpansion& output_arg : output_arg_expansions_) { in GrapplerFunctionItem() local
334 for (const string& output_node : output_arg.output_nodes) { in GrapplerFunctionItem()
403 const auto& output = func.signature().output_arg(); in HasParametrizedType()
445 for (const auto& output : func.signature().output_arg()) in InstantiationTypeParameters()
531 for (const OpDef::ArgDef& output : signature.output_arg()) { in MakeGrapplerFunctionItem()
606 for (const OpDef::ArgDef& out : signature.output_arg()) { in MakeGrapplerFunctionItem()
794 for (const OutputArgExpansion& output_arg : item.outputs()) { in MakeFunctionDef() local
795 for (const string& output_node : output_arg.output_nodes) { in MakeFunctionDef()
824 [&output_tensors](const OutputArgExpansion& output_arg) -> const string& { in MakeFunctionDef() argument
825 const string& output_node = output_arg.output_nodes[0]; in MakeFunctionDef()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/ |
| D | function_functional_while.pbtxt | 9 output_arg {
13 output_arg {
125 output_arg {
129 output_arg {
199 output_arg {
|
| D | simple_function.pbtxt | 79 output_arg {
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 58 for (const auto& output_arg : op_def.output_arg()) { in GetNumRetvals() local
59 if (!output_arg.number_attr().empty()) { in GetNumRetvals()
60 auto iter = attrs.find(output_arg.number_attr()); in GetNumRetvals()
63 output_arg.number_attr(), in GetNumRetvals()
67 } else if (!output_arg.type_list_attr().empty()) { in GetNumRetvals()
68 auto iter = attrs.find(output_arg.type_list_attr()); in GetNumRetvals()
71 output_arg.type_list_attr(), in GetNumRetvals()
|
| /external/tensorflow/tensorflow/go/genop/internal/ |
| D | genop_test.go | 94 output_arg: <
143 output_arg: <
189 output_arg: <
316 output_arg <
320 output_arg <
361 output_arg: <
453 output_arg: <
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_to_function_def.py | 83 for arg_def in op_def.output_arg:
159 func.signature.output_arg.extend(
171 func.signature.output_arg.extend(
184 k = func.signature.output_arg[index].name
|
| D | python_op_gen_internal.cc | 289 StringPiece description = op_def.output_arg(0).description(); in GetReturns()
302 desc = op_def.output_arg(0).description(); in GetReturns()
303 } else if (!op_def.output_arg(0).name().empty()) { in GetReturns()
304 desc = strings::StrCat(" The ", op_def.output_arg(0).name(), in GetReturns()
315 if (!op_def.output_arg(i).name().empty()) { in GetReturns()
316 out_names[i] = op_def.output_arg(i).name(); in GetReturns()
325 StringPiece description = op_def.output_arg(i).description(); in GetReturns()
801 ArgTypeName(op_def_, op_def_.output_arg(i), inferred_attrs_, true)); in AddDocStringOutputs()
|
| D | function_def_to_graph.py | 71 for arg in fdef.signature.output_arg
175 for arg_def in op_def.output_arg:
|
| D | function.py | 394 self._definition.signature.output_arg, self._definition.node_def)
452 def _create_hash_str(self, input_arg, output_arg, node_def): argument
483 for adef in output_arg:
616 … dtypes.DType(_.type) for _ in defined._signature.output_arg # pylint: disable=protected-access
1015 output_types = [dtypes.DType(x.type) for x in sig.output_arg]
1058 out_names = [arg.name for arg in fdef.signature.output_arg]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | graph_rewrite_dataset.cc | 103 for (const auto& output : function_def->signature().output_arg()) { in AddFakeSinks()
124 for (const auto& output_arg : function_def->signature().output_arg()) { in RemoveFakeSinks() local
125 const string& tensor = function_def->ret().at(output_arg.name()); in RemoveFakeSinks()
128 (*function_def->mutable_ret())[output_arg.name()] = in RemoveFakeSinks()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info.cc | 53 for (const auto& output_arg : function_def.signature().output_arg()) { in Init() local
54 output_arg_dtypes_.emplace_back(output_arg.type()); in Init()
117 if (!IsSameArgDef(sig1.output_arg(k), sig2.output_arg(k))) return false; in IsSameSignature()
|
| /external/tensorflow/tensorflow/python/framework/testdata/ |
| D | metrics_export_meta_graph.pb | 14 output_arg {
50 output_arg {
91 output_arg {
106 output_arg {
121 output_arg {
173 output_arg {
201 output_arg {
219 output_arg {
273 output_arg {
311 output_arg {
[all …]
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/ |
| D | saved_model.pbtxt | 15 output_arg {
50 output_arg {
77 output_arg {
96 output_arg {
133 output_arg {
168 output_arg {
214 output_arg {
219 output_arg {
223 output_arg {
228 output_arg {
[all …]
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_graphs.py | 325 for i, output_arg in enumerate(op_def.output_arg):
326 if output_arg.is_ref:
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_operation_table_test.cc | 33 absl::c_any_of(op_def.output_arg(), IsResourceArgDef); in HasResourceInputOrOutput()
|