Home
last modified time | relevance | path

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

12345

/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/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/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/optimizers/
Dremapper.cc502 fused_conv2d->add_input(matched.conv2d->input(0)); // 0: input in AddFusedConv2DNode()
503 fused_conv2d->add_input(matched.conv2d->input(1)); // 1: filter in AddFusedConv2DNode()
504 fused_conv2d->add_input(matched.bias_add->input(1)); // 2: bias in AddFusedConv2DNode()
529 fused_conv2d->add_input(matched.conv2d->input(0)); // 0: input in AddFusedConv2DNode()
530 fused_conv2d->add_input(matched.conv2d->input(1)); // 1: filter in AddFusedConv2DNode()
531 fused_conv2d->add_input(matched.bias_add->input(1)); // 2: bias in AddFusedConv2DNode()
559 fused_conv2d->add_input(matched.conv2d->input(0)); // 0: input in AddFusedConv2DNode()
560 fused_conv2d->add_input(matched.conv2d->input(1)); // 1: filter in AddFusedConv2DNode()
561 fused_conv2d->add_input(matched.bias_add->input(1)); // 2: bias in AddFusedConv2DNode()
588 fused_conv2d->add_input(matched.conv2d->input(0)); // 0: input in AddFusedConv2DNode()
[all …]
Dmemory_optimizer.cc341 *new_trigger_node->add_input() = in AddRecomputeControlDependencyNodes()
352 *current_trigger_node->add_input() = in AddRecomputeControlDependencyNodes()
405 *copied_node->add_input() = RecomputedOrOriginalNodeName( in RecomputeSubgraph()
410 *copied_node->add_input() = in RecomputeSubgraph()
651 *tmp_var->add_input() = ctrl_dep; in SchedulingPass()
653 *tmp_var->add_input() = in SchedulingPass()
662 *zeros->add_input() = node->input(min_input_id); in SchedulingPass()
671 *initialize->add_input() = tmp_var->name(); in SchedulingPass()
672 *initialize->add_input() = zeros->name(); in SchedulingPass()
686 *accumulate->add_input() = initialize->name(); in SchedulingPass()
[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.cc51 new_node.add_input(shuffle_node.input(0)); in OptimizeAndCollectStats()
54 new_node.add_input(shuffle_node.input(1)); in OptimizeAndCollectStats()
57 new_node.add_input(shuffle_node.input(2)); in OptimizeAndCollectStats()
60 new_node.add_input(shuffle_node.input(3)); in OptimizeAndCollectStats()
63 new_node.add_input(repeat_node.input(1)); in OptimizeAndCollectStats()
Dmap_vectorization.cc94 map_defun_node->add_input(input.name()); in CreateMapDefunWrapper()
170 to->add_input(from.input(i)); in CopyInputs()
200 batch_node.add_input(input_node.name()); in AddNewBatchNode()
259 map_node.add_input(new_batch_node.name()); in AddNewMapNode()
276 map_node.add_input(autotune_val->name()); in AddNewMapNode()
303 prefetch_node.add_input(new_map_node.name()); in AddNewPrefetchNode()
310 prefetch_node.add_input(autotune_val->name()); in AddNewPrefetchNode()
352 function_node->add_input(prev_node_output); in AddBranch()
364 choose_fastest_node->add_input(node->input(i)); in AddBranch()
371 function_node->add_input(input_arg_name); in AddBranch()
[all …]
/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()
106 async_known_many->add_input(source1); in TEST_P()
109 async_known_many->add_input(source2); in TEST_P()
170 interleave_many->add_input(meta_source); in TEST()
173 interleave_many->add_input(source1); in TEST()
176 interleave_many->add_input(source2); in TEST()
207 known_many->add_input(source1); in TEST_P()
210 known_many->add_input(source2); in TEST_P()
[all …]
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc258 *r->add_input() = "shape"; in RunInfiniteTFLoop()
265 *e->add_input() = "random"; in RunInfiniteTFLoop()
272 *m->add_input() = "while/Enter"; in RunInfiniteTFLoop()
273 *m->add_input() = "while/NextIteration"; in RunInfiniteTFLoop()
279 *t->add_input() = "^while/Merge"; in RunInfiniteTFLoop()
288 *c->add_input() = "always_true"; in RunInfiniteTFLoop()
294 *s->add_input() = "while/Merge"; in RunInfiniteTFLoop()
295 *s->add_input() = "while/LoopCond"; in RunInfiniteTFLoop()
301 *i->add_input() = "while/Switch:1"; in RunInfiniteTFLoop()
307 *n->add_input() = "while/Identity"; in RunInfiniteTFLoop()
[all …]
/external/mesa3d/src/compiler/glsl/
Dbuiltin_variables.cpp383 ir_variable *add_input(int slot, const glsl_type *type, const char *name) in add_input() function in __anon72aa8a280111::builtin_variable_generator
1039 add_input(VERT_ATTRIB_POS, vec4_t, "gl_Vertex"); in generate_vs_special_vars()
1040 add_input(VERT_ATTRIB_NORMAL, vec3_t, "gl_Normal"); in generate_vs_special_vars()
1041 add_input(VERT_ATTRIB_COLOR0, vec4_t, "gl_Color"); in generate_vs_special_vars()
1042 add_input(VERT_ATTRIB_COLOR1, vec4_t, "gl_SecondaryColor"); in generate_vs_special_vars()
1043 add_input(VERT_ATTRIB_TEX0, vec4_t, "gl_MultiTexCoord0"); in generate_vs_special_vars()
1044 add_input(VERT_ATTRIB_TEX1, vec4_t, "gl_MultiTexCoord1"); in generate_vs_special_vars()
1045 add_input(VERT_ATTRIB_TEX2, vec4_t, "gl_MultiTexCoord2"); in generate_vs_special_vars()
1046 add_input(VERT_ATTRIB_TEX3, vec4_t, "gl_MultiTexCoord3"); in generate_vs_special_vars()
1047 add_input(VERT_ATTRIB_TEX4, vec4_t, "gl_MultiTexCoord4"); in generate_vs_special_vars()
[all …]
/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()
153 node.add_input(input_node_name); in CreateFaninFanoutNodeGraph()
158 output_node.add_input(absl::StrCat(node.name(), ":", i)); in CreateFaninFanoutNodeGraph()
/external/tensorflow/tensorflow/lite/python/
Dconvert_test.py324 a = custom.add_input(a, tag="mytag",
327 c = custom.add_input(c, tag="mytag",
329 d = custom.add_input(d, tag="mytag2",
346 b = custom.add_input(b) # should auto assign 0
347 a = custom.add_input(a, index_override=1)
348 c = custom.add_input(c) # should auto assign 2
361 a0 = hint.add_input(a0, tag="c", aggregate=op_hint.OpHint.AGGREGATE_STACK)
362 b0 = hint.add_input(b0, tag="n", aggregate=op_hint.OpHint.AGGREGATE_STACK)
363 a1 = hint.add_input(a1, tag="c", aggregate=op_hint.OpHint.AGGREGATE_STACK)
364 b1 = hint.add_input(b1, tag="n", aggregate=op_hint.OpHint.AGGREGATE_STACK)
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_reachability.cc86 const auto add_input = [&channel_group, &inputs](HloInstruction* input) { in Build() local
96 const auto add_dependencies = [&add_input](const HloInstruction* hlo) { in Build()
98 add_input(operand); in Build()
101 add_input(predecessor); in Build()
113 add_input(channel); in Build()
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py111 return self._tflite_wrapper.add_input(
128 inputs = self._tflite_wrapper.add_input(
130 state = self._tflite_wrapper.add_input(
313 return self._tflite_wrapper.add_input(
409 inputs = self._tflite_wrapper.add_input(
426 c_prev = self._tflite_wrapper.add_input(
432 m_prev = self._tflite_wrapper.add_input(

12345