Home
last modified time | relevance | path

Searched refs:add_input (Results 1 – 25 of 119) sorted by relevance

12345

/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_nodes_test.cc44 add_node1->add_input("add_node2"); in TestRemoveNodes()
45 add_node1->add_input("add_node3"); in TestRemoveNodes()
50 add_node2->add_input("identity_node1"); in TestRemoveNodes()
51 add_node2->add_input("identity_node2"); in TestRemoveNodes()
56 add_node3->add_input("identity_node1"); in TestRemoveNodes()
57 add_node3->add_input("const_node3"); in TestRemoveNodes()
62 identity_node1->add_input("const_node1"); in TestRemoveNodes()
67 identity_node2->add_input("const_node2"); in TestRemoveNodes()
72 identity_node3->add_input("const_node3"); in TestRemoveNodes()
89 add_node4->add_input("add_node2"); in TestRemoveNodes()
[all …]
Dsort_by_execution_order_test.cc45 add_node->add_input("a_node"); in TestSimpleAdd()
46 add_node->add_input("b_node"); in TestSimpleAdd()
72 negative_node->add_input("sqrt_node"); in TestSimpleLinear()
77 relu_node->add_input("const_node"); in TestSimpleLinear()
82 sqrt_node->add_input("relu_node"); in TestSimpleLinear()
105 add_node1->add_input("add_node2"); in TestSimpleTree()
106 add_node1->add_input("add_node3"); in TestSimpleTree()
111 add_node2->add_input("const_node1"); in TestSimpleTree()
112 add_node2->add_input("const_node2"); in TestSimpleTree()
117 add_node3->add_input("const_node3"); in TestSimpleTree()
[all …]
Dstrip_unused_nodes_test.cc44 add_node->add_input("a_node"); in TestSimpleAdd()
45 add_node->add_input("b_node"); in TestSimpleAdd()
76 add_node1->add_input("add_node2"); in TestCommonAncestor()
77 add_node1->add_input("add_node3"); in TestCommonAncestor()
82 add_node2->add_input("const_node1"); in TestCommonAncestor()
83 add_node2->add_input("const_node2"); in TestCommonAncestor()
88 add_node3->add_input("const_node1"); in TestCommonAncestor()
89 add_node3->add_input("const_node3"); in TestCommonAncestor()
110 add_node4->add_input("add_node2"); in TestCommonAncestor()
111 add_node4->add_input("add_node3"); in TestCommonAncestor()
[all …]
Dbackports_test.cc48 mul_node1->add_input("add_node2"); in TestBackportConcatV2()
49 mul_node1->add_input("add_node3"); in TestBackportConcatV2()
54 add_node2->add_input("const_node1"); in TestBackportConcatV2()
55 add_node2->add_input("const_node2"); in TestBackportConcatV2()
60 add_node3->add_input("const_node1"); in TestBackportConcatV2()
61 add_node3->add_input("const_node3"); in TestBackportConcatV2()
78 concat_node->add_input("const_node1"); in TestBackportConcatV2()
79 concat_node->add_input("const_node2"); in TestBackportConcatV2()
80 concat_node->add_input("const_node3"); in TestBackportConcatV2()
121 tensor_array_node->add_input("size_node"); in TEST()
[all …]
Dremove_device_test.cc45 mul_node1->add_input("add_node2"); in TestRemoveDevice()
46 mul_node1->add_input("add_node3"); in TestRemoveDevice()
51 add_node2->add_input("const_node1"); in TestRemoveDevice()
52 add_node2->add_input("const_node2"); in TestRemoveDevice()
58 add_node3->add_input("const_node1"); in TestRemoveDevice()
59 add_node3->add_input("const_node3"); in TestRemoveDevice()
76 add_node4->add_input("add_node2"); in TestRemoveDevice()
77 add_node4->add_input("add_node3"); in TestRemoveDevice()
Drename_op_test.cc44 mul_node1->add_input("add_node2"); in TestRenameOp()
45 mul_node1->add_input("add_node3"); in TestRenameOp()
50 add_node2->add_input("const_node1"); in TestRenameOp()
51 add_node2->add_input("const_node2"); in TestRenameOp()
56 add_node3->add_input("const_node1"); in TestRenameOp()
57 add_node3->add_input("const_node3"); in TestRenameOp()
74 add_node4->add_input("add_node2"); in TestRenameOp()
75 add_node4->add_input("add_node3"); in TestRenameOp()
Dset_device_test.cc44 mul_node1->add_input("add_node2"); in CreateDeviceGraph()
45 mul_node1->add_input("add_node3"); in CreateDeviceGraph()
50 add_node2->add_input("const_node1"); in CreateDeviceGraph()
51 add_node2->add_input("const_node2"); in CreateDeviceGraph()
57 add_node3->add_input("const_node1"); in CreateDeviceGraph()
58 add_node3->add_input("const_node3"); in CreateDeviceGraph()
75 add_node4->add_input("add_node2"); in CreateDeviceGraph()
76 add_node4->add_input("add_node3"); in CreateDeviceGraph()
Dremove_attribute_test.cc44 mul_node1->add_input("add_node2"); in TestRemoveAttribute()
45 mul_node1->add_input("add_node3"); in TestRemoveAttribute()
52 add_node2->add_input("const_node1"); in TestRemoveAttribute()
53 add_node2->add_input("const_node2"); in TestRemoveAttribute()
61 add_node3->add_input("const_node1"); in TestRemoveAttribute()
62 add_node3->add_input("const_node3"); in TestRemoveAttribute()
79 add_node4->add_input("add_node2"); in TestRemoveAttribute()
80 add_node4->add_input("add_node3"); in TestRemoveAttribute()
Drename_attribute_test.cc44 mul_node1->add_input("add_node2"); in TestRenameAttribute()
45 mul_node1->add_input("add_node3"); in TestRenameAttribute()
52 add_node2->add_input("const_node1"); in TestRenameAttribute()
53 add_node2->add_input("const_node2"); in TestRenameAttribute()
61 add_node3->add_input("const_node1"); in TestRenameAttribute()
62 add_node3->add_input("const_node3"); in TestRenameAttribute()
79 add_node4->add_input("add_node2"); in TestRenameAttribute()
80 add_node4->add_input("add_node3"); in TestRenameAttribute()
Dobfuscate_names_test.cc44 add_node1->add_input("add_node2"); in TestSimpleTree()
45 add_node1->add_input("add_node3"); in TestSimpleTree()
50 add_node2->add_input("const_node1"); in TestSimpleTree()
51 add_node2->add_input("const_node2"); in TestSimpleTree()
56 add_node3->add_input("const_node3"); in TestSimpleTree()
57 add_node3->add_input("const_node4"); in TestSimpleTree()
/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
Dtest_tools.cc40 n.add_input(arg1); in MakeNode2Arg()
41 n.add_input(arg2); in MakeNode2Arg()
51 n.add_input(arg1); in MakeNode4Arg()
52 n.add_input(arg2); in MakeNode4Arg()
53 n.add_input(arg3); in MakeNode4Arg()
54 n.add_input(arg4); in MakeNode4Arg()
157 node3->add_input("^node3"); // The control link goes back to self. in TestGraphs()
173 add2->add_input("const2_3"); in TestGraphs()
174 add2->add_input("const2_3"); in TestGraphs()
193 pass2->add_input("const2_3"); in TestGraphs()
[all …]
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc395 *conv2d_op->add_input() = src_op.inputs[0]; in ConvertConvOperator()
396 *conv2d_op->add_input() = src_op.inputs[1]; in ConvertConvOperator()
431 biasadd_op->add_input(conv_output); in ConvertConvOperator()
432 biasadd_op->add_input(src_op.inputs[2]); in ConvertConvOperator()
463 *dc2d_op->add_input() = src_op.inputs[0]; in ConvertDepthwiseConvOperator()
464 *dc2d_op->add_input() = src_op.inputs[1]; in ConvertDepthwiseConvOperator()
522 biasadd_op->add_input(conv_output); in ConvertDepthwiseConvOperator()
523 biasadd_op->add_input(src_op.inputs[2]); in ConvertDepthwiseConvOperator()
547 *conv2d_op->add_input() = src_op.inputs[0]; in ConvertTransposeConvOperator()
548 *conv2d_op->add_input() = src_op.inputs[1]; in ConvertTransposeConvOperator()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dcanonicalizer_test.cc28 *node.add_input() = "b"; in MakeNode()
29 *node.add_input() = "a"; in MakeNode()
30 *node.add_input() = "^z"; in MakeNode()
31 *node.add_input() = "^y"; in MakeNode()
32 *node.add_input() = "^x"; in MakeNode()
33 *node.add_input() = "^z"; in MakeNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc920 fused_op.add_input(contraction.input(0)); // 0: input in AddFusedContractionNode()
921 fused_op.add_input(contraction.input(1)); // 1: filter in AddFusedContractionNode()
922 fused_op.add_input(bias_add.input(1)); // 2: bias in AddFusedContractionNode()
964 fused_op.add_input(contraction.input(0)); // 0: input in AddFusedContractionNode()
965 fused_op.add_input(contraction.input(1)); // 1: filter in AddFusedContractionNode()
966 fused_op.add_input(bias_add.input(1)); // 2: bias in AddFusedContractionNode()
1013 fused_conv2d.add_input(contraction.input(0)); // 0: input in AddFusedConv2DNode()
1014 fused_conv2d.add_input(contraction.input(1)); // 1: filter in AddFusedConv2DNode()
1015 fused_conv2d.add_input(bias_add.input(1)); // 2: bias in AddFusedConv2DNode()
1055 fused_conv2d.add_input(contraction.input(0)); // 0: input in AddFusedConv2DNode()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_batch_fusion.cc43 new_node.add_input(map_node.input(0)); in MakeMapAndBatchNode()
53 new_node.add_input(map_node.input(i + 1)); in MakeMapAndBatchNode()
57 new_node.add_input(batch_node.input(1)); in MakeMapAndBatchNode()
67 new_node.add_input(tmp->name()); in MakeMapAndBatchNode()
70 new_node.add_input(tmp->name()); in MakeMapAndBatchNode()
75 new_node.add_input(batch_node.input(2)); in MakeMapAndBatchNode()
78 new_node.add_input(tmp->name()); in MakeMapAndBatchNode()
Dhoist_random_uniform.cc47 stateless_map.add_input(zip_node.name()); in MakeStatelessMap()
50 stateless_map.add_input(map_node.input(i)); in MakeStatelessMap()
80 random_dataset.add_input(seed->name()); in MakeRandomDataset()
81 random_dataset.add_input(seed2->name()); in MakeRandomDataset()
97 batch_dataset.add_input(random_dataset.name()); in MakeBatchTwo()
98 batch_dataset.add_input(batch_size->name()); in MakeBatchTwo()
99 batch_dataset.add_input(drop_reminder->name()); in MakeBatchTwo()
120 zip_node.add_input(first_node.name()); in MakeZipNode()
121 zip_node.add_input(second_node.name()); in MakeZipNode()
169 random_uniform->add_input(seed_arg->name()); in MakeLessStatefulFunction()
Dshuffle_and_repeat_fusion.cc52 new_node.add_input(shuffle_node.input(0)); in OptimizeAndCollectStats()
55 new_node.add_input(shuffle_node.input(1)); in OptimizeAndCollectStats()
58 new_node.add_input(shuffle_node.input(2)); in OptimizeAndCollectStats()
61 new_node.add_input(shuffle_node.input(3)); in OptimizeAndCollectStats()
64 new_node.add_input(repeat_node.input(1)); in OptimizeAndCollectStats()
Dmap_vectorization.cc95 map_defun_node->add_input(input.name()); in CreateMapDefunWrapper()
171 to->add_input(from.input(i)); in CopyInputs()
201 batch_node.add_input(input_node.name()); in AddNewBatchNode()
260 map_node.add_input(new_batch_node.name()); in AddNewMapNode()
277 map_node.add_input(autotune_val->name()); in AddNewMapNode()
304 prefetch_node.add_input(new_map_node.name()); in AddNewPrefetchNode()
311 prefetch_node.add_input(autotune_val->name()); in AddNewPrefetchNode()
353 function_node->add_input(prev_node_output); in AddBranch()
365 choose_fastest_node->add_input(node->input(i)); in AddBranch()
372 function_node->add_input(input_arg_name); in AddBranch()
[all …]
Dauto_shard.cc135 new_node.add_input(add_before.input(0)); in AddShardNode()
136 new_node.add_input(num_shards_node->name()); in AddShardNode()
137 new_node.add_input(index_node->name()); in AddShardNode()
197 new_node.add_input(add_before.input(0)); in AddShuffleNode()
198 new_node.add_input(buffer_size_node); in AddShuffleNode()
199 new_node.add_input(seed_node); in AddShuffleNode()
200 new_node.add_input(seed2_node); in AddShuffleNode()
225 new_node.add_input(add_before.input(0)); in AddShuffleV2Node()
226 new_node.add_input(buffer_size_node); in AddShuffleV2Node()
227 new_node.add_input(seed_generator_node); in AddShuffleV2Node()
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h95 node.add_input(absl::StrFormat("in%04d", rnd.Uniform(kNumInNodes))); in CreateGraphDef()
104 node.add_input(absl::StrFormat("op%05d", op)); in CreateGraphDef()
129 node.add_input(absl::StrCat(prefix, input_node)); in CreateRandomGraph()
157 node.add_input(input_node_name); in CreateFaninFanoutNodeGraph()
164 node.add_input(absl::StrCat("^", input_node_name)); in CreateFaninFanoutNodeGraph()
171 output_node.add_input(input_node_index); in CreateFaninFanoutNodeGraph()
179 output_node.add_input(controlled_fanout_input); in CreateFaninFanoutNodeGraph()
/external/tensorflow/tensorflow/core/framework/
Dmodel_test.cc42 async_interleave_many->add_input(meta_source); in TEST_P()
48 async_interleave_many->add_input(source1); in TEST_P()
54 async_interleave_many->add_input(source2); in TEST_P()
122 async_known_many->add_input(source1); in TEST_P()
125 async_known_many->add_input(source2); in TEST_P()
195 interleave_many->add_input(meta_source); in TEST()
198 interleave_many->add_input(source1); in TEST()
201 interleave_many->add_input(source2); in TEST()
237 known_many->add_input(source1); in TEST_P()
240 known_many->add_input(source2); in TEST_P()
[all …]
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc266 *r->add_input() = "shape"; in RunInfiniteTFLoop()
273 *e->add_input() = "random"; in RunInfiniteTFLoop()
280 *m->add_input() = "while/Enter"; in RunInfiniteTFLoop()
281 *m->add_input() = "while/NextIteration"; in RunInfiniteTFLoop()
287 *t->add_input() = "^while/Merge"; in RunInfiniteTFLoop()
296 *c->add_input() = "always_true"; in RunInfiniteTFLoop()
302 *s->add_input() = "while/Merge"; in RunInfiniteTFLoop()
303 *s->add_input() = "while/LoopCond"; in RunInfiniteTFLoop()
309 *i->add_input() = "while/Switch:1"; in RunInfiniteTFLoop()
315 *n->add_input() = "while/Identity"; in RunInfiniteTFLoop()
[all …]
/external/tensorflow/tensorflow/lite/python/
Dconvert_test.py336 a = custom.add_input(
339 c = custom.add_input(
341 d = custom.add_input(
359 b = custom.add_input(b) # should auto assign 0
360 a = custom.add_input(a, index_override=1)
361 c = custom.add_input(c) # should auto assign 2
375 a0 = hint.add_input(a0, tag="c", aggregate=op_hint.OpHint.AGGREGATE_STACK)
376 b0 = hint.add_input(b0, tag="n", aggregate=op_hint.OpHint.AGGREGATE_STACK)
377 a1 = hint.add_input(a1, tag="c", aggregate=op_hint.OpHint.AGGREGATE_STACK)
378 b1 = hint.add_input(b1, tag="n", aggregate=op_hint.OpHint.AGGREGATE_STACK)
[all …]
/external/mesa3d/src/compiler/glsl/
Dbuiltin_variables.cpp400 ir_variable *add_input(int slot, const glsl_type *type, int precision, in add_input() function in __anon9506292a0111::builtin_variable_generator
407 ir_variable *add_input(int slot, const glsl_type *type, const char *name, in add_input() function in __anon9506292a0111::builtin_variable_generator
410 return add_input(slot, type, GLSL_PRECISION_NONE, name, interp); in add_input()
1110 add_input(VERT_ATTRIB_POS, vec4_t, "gl_Vertex"); in generate_vs_special_vars()
1111 add_input(VERT_ATTRIB_NORMAL, vec3_t, "gl_Normal"); in generate_vs_special_vars()
1112 add_input(VERT_ATTRIB_COLOR0, vec4_t, "gl_Color"); in generate_vs_special_vars()
1113 add_input(VERT_ATTRIB_COLOR1, vec4_t, "gl_SecondaryColor"); in generate_vs_special_vars()
1114 add_input(VERT_ATTRIB_TEX0, vec4_t, "gl_MultiTexCoord0"); in generate_vs_special_vars()
1115 add_input(VERT_ATTRIB_TEX1, vec4_t, "gl_MultiTexCoord1"); in generate_vs_special_vars()
1116 add_input(VERT_ATTRIB_TEX2, vec4_t, "gl_MultiTexCoord2"); in generate_vs_special_vars()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_reachability.cc95 const auto add_input = [&channel_group, &inputs](HloInstruction* input) { in Build() local
105 const auto add_dependencies = [&add_input](const HloInstruction* hlo) { in Build()
107 add_input(operand); in Build()
110 add_input(predecessor); in Build()
124 add_input(channel); in Build()

12345