Home
last modified time | relevance | path

Searched refs:AddN (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer_test.cc48 Output d = ops::AddN(s.WithOpName("gradients/d"), {c}); in TEST_F()
49 Output e = ops::AddN(s.WithOpName("gradients/e"), {d, b}); in TEST_F()
50 Output f = ops::AddN(s.WithOpName("gradients/f"), {e, a}); in TEST_F()
87 Output d = ops::AddN(s.WithOpName("gradients/d"), {c}); in TEST_F()
88 Output e = ops::AddN(s.WithOpName("gradients/e"), {d, b}); in TEST_F()
89 Output f = ops::AddN(s.WithOpName("gradients/f"), {e, a}); in TEST_F()
112 Output d = ops::AddN( in TEST_F()
149 Output trigger = ops::AddN(s.WithOpName("gradients/BN1Grad"), {d}); in TEST_F()
150 Output e = ops::AddN(s.WithOpName("gradients/Conv1Grad"), {trigger, c}); in TEST_F()
151 Output f = ops::AddN(s.WithOpName("gradients/ReLUGrad"), {e, c}); in TEST_F()
[all …]
Dstatic_schedule_test.cc87 Output b = ops::AddN(s.WithOpName("b"), {a}); in TEST_F()
90 Output e = ops::AddN(s.WithOpName("e"), {d}); in TEST_F()
Dmodel_pruner_test.cc259 Output b = ops::AddN(s.WithOpName("b"), {a}); in TEST_F()
260 Output c = ops::AddN(s.WithOpName("c"), {b}); in TEST_F()
261 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(b), {c}); in TEST_F()
262 Output e = ops::AddN(s.WithOpName("e"), {d}); in TEST_F()
Dauto_parallel_test.cc40 Output add = ops::AddN(s.WithOpName("add"), {constant_a, dequeue[0]}); in TEST_F()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc104 outputs->push_back(ops::AddN(s, {inputs[0], 1})); in TEST_F()
123 outputs->push_back(ops::AddN(s, {inputs[0], inputs[1]})); in TEST_F()
124 outputs->push_back(ops::AddN(s, {inputs[1], 1})); in TEST_F()
175 outputs->push_back(ops::AddN(s, {inputs[0], -1.0})); in TEST_F()
205 outputs->push_back(ops::AddN(cpu1_scope, {inputs[0], inputs[1]})); in TEST_F()
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_AddN.pbtxt2 graph_op_name: "AddN"
4 name: "math.AddN"
/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.cc58 AddN(s.WithDevice( in CreateGraphDef()
81 AddN output(s.WithOpName("y").WithDevice("/CPU:0"), last_stage); in CreateGraphDef()
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_AddN.pbtxt2 graph_op_name: "AddN"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AddN.pbtxt2 graph_op_name: "AddN"
Dapi_def_AddV2.pbtxt5 *NOTE*: `Add` supports broadcasting. `AddN` does not. More about broadcasting
Dapi_def_Add.pbtxt5 *NOTE*: `Add` supports broadcasting. `AddN` does not. More about broadcasting
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc141 Output combine = AddN( in CreateGraphDef()
150 /* Output y =*/AddN(s.WithOpName("y"), last_stage); in CreateGraphDef()
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n.cc59 reference_ops::AddN<T>(GetTensorShape(input1), num_inputs, all_inputs.data(), in EvalAddN()
/external/tensorflow/tensorflow/core/grappler/
Dgraph_view_test.cc185 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(a), {b, c}); in TEST_F()
249 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(a), {b, c}); in TEST_F()
271 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(a), {b, c}); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test_test.cc87 auto add_all = ops::AddN(s.WithOpName("Add_all"), inputs); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef_test.cc62 auto g = ops::AddN(root.WithOpName("G"), std::initializer_list<Output>{e, f}); in TEST()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc44 auto g = ops::AddN(root.WithOpName("G"), std::initializer_list<Output>{e, f}); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dcontrol_flow_test.cc37 outputs->push_back(ops::AddN(scope, {inputs[0], 1})); in AddOneBody()
Dgraph_partition_test.cc426 outputs->push_back(ops::AddN(cpu1, {inputs[0], inputs[1]})); in TEST_F()
585 ops::AddN addn = ops::AddN(root.WithOpName("addn"), in TEST()
/external/tensorflow/tensorflow/core/profiler/g3doc/
Dprofile_memory.md36 AddN 2741.49MB (8.56%, 2.28%)
Dprofile_time.md130 AddN 50.10ms (17.33%, 1.34%), 5481
/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad_test.cc32 using ops::AddN;
801 TEST_F(NaryGradTest, AddN) { in TEST_F() argument
807 auto y = AddN(scope_, xs); in TEST_F()
/external/tensorflow/tensorflow/go/op/
Dgradients_test.go33 y2 = AddN(s.SubScope("y2"), []tf.Output{y0, x2})
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc1818 auto o = ops::AddN(scope.WithOpName("o"), x.output); in TEST()
1838 auto o = ops::AddN(scope.WithOpName("o"), in TEST()
1854 ops::AddN(scope.WithOpName("o"), std::initializer_list<Input>{l, r}); in TEST()
1899 ops::AddN(s.WithOpName("o").WithControlDependencies({x.output[0].op()}), in TEST()
1921 auto o = ops::AddN(s.WithOpName("o").WithControlDependencies({func_3}), in TEST()
/external/tensorflow/tensorflow/core/protobuf/
Drewriter_config.proto116 // Scheduling will split big ops such as AddN and try to enforce a schedule

12