Home
last modified time | relevance | path

Searched refs:WithOpName (Results 1 – 25 of 134) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc107 auto x = ops::Placeholder(scope.WithOpName("x"), DT_INT32); in BuildCondGraph()
108 auto y = ops::Placeholder(scope.WithOpName("y"), DT_INT32); in BuildCondGraph()
109 auto less = ops::Less(scope.WithOpName("cond/Less"), y, x); in BuildCondGraph()
110 auto switch_1 = ops::Switch(scope.WithOpName("cond/Switch"), less, less); in BuildCondGraph()
113 ops::Identity(scope.WithOpName("cond/Identity"), switch_1.output_true); in BuildCondGraph()
115 scope.WithOpName("cond").WithControlDependencies(identity_t), 17); in BuildCondGraph()
116 auto switch_2 = ops::Switch(scope.WithOpName("cond/Switch"), y, less); in BuildCondGraph()
117 auto mul = ops::Multiply(scope.WithOpName("cond/Mul"), switch_2.output_true, in BuildCondGraph()
121 ops::Identity(scope.WithOpName("cond/Identity"), switch_1.output_false); in BuildCondGraph()
123 scope.WithOpName("cond").WithControlDependencies(identity_f), 23); in BuildCondGraph()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner_test.cc59 Output a = ops::Const(s.WithOpName("a"), 0.0f, {10, 10}); in TEST_F()
60 Output b = ops::Sqrt(s.WithOpName("b"), {a}); in TEST_F()
61 Output c = ops::StopGradient(s.WithOpName("c"), b); in TEST_F()
62 Output d = ops::StopGradient(s.WithOpName("d"), c); in TEST_F()
63 Output e = ops::Sqrt(s.WithOpName("e"), {d}); in TEST_F()
76 Output a = ops::Const(s.WithOpName("a"), 0.0f, {10, 10}); in TEST_F()
77 Output b = ops::Sqrt(s.WithOpName("b"), {a}); in TEST_F()
78 Output c = ops::StopGradient(s.WithOpName("c"), b); in TEST_F()
79 Output d = ops::StopGradient(s.WithOpName("d"), b); in TEST_F()
80 Output e = ops::Sqrt(s.WithOpName("e"), {b}); in TEST_F()
[all …]
Darithmetic_optimizer_test.cc111 Output c = ops::Const(s.WithOpName("c"), {1.0f, 2.0f}, {1, 2}); in TEST_F()
112 Output d = ops::Const(s.WithOpName("d"), {3.0f, 4.0f}, {1, 2}); in TEST_F()
113 Output mul = ops::Mul(s.WithControlDependencies(d).WithOpName("mul"), c, c); in TEST_F()
114 Output mul_no_nan = ops::MulNoNan(s.WithOpName("mul_no_nan"), d, d); in TEST_F()
115 Output id = ops::Identity(s.WithOpName("id"), mul); in TEST_F()
116 Output id2 = ops::Identity(s.WithOpName("id2"), mul_no_nan); in TEST_F()
156 auto c = ops::Const(s.WithOpName("c"), {1.0f, 2.0f}, {1, 2}); in TEST_F()
157 auto neg1 = ops::Neg(s.WithOpName("neg1"), c); in TEST_F()
158 auto neg2 = ops::Neg(s.WithOpName("neg2"), neg1); in TEST_F()
159 auto recip1 = ops::Reciprocal(s.WithOpName("recip1"), neg2); in TEST_F()
[all …]
Ddependency_optimizer_test.cc69 Output x = ops::Const(s.WithOpName("x"), {1.0f, 2.0f}, {1, 2}); in TEST_F()
70 Output y = ops::Const(s.WithOpName("y"), {1.0f, 2.0f}, {1, 2}); in TEST_F()
71 Output z = ops::Const(s.WithOpName("z"), {1.0f, 2.0f}, {1, 2}); in TEST_F()
72 Output add = ops::Add(s.WithOpName("add"), x, y); in TEST_F()
74 ops::Identity(s.WithOpName("id1").WithControlDependencies(x), add); in TEST_F()
76 s.WithOpName("id2").WithControlDependencies(y).WithControlDependencies(z), in TEST_F()
106 Output x = ops::RandomUniform(s.WithOpName("x"), {1, 2}, DT_FLOAT); in TEST_F()
107 Output y = ops::RandomUniform(s.WithOpName("y"), {1, 2}, DT_FLOAT); in TEST_F()
108 Output add = ops::Add(s.WithOpName("add"), x, y); in TEST_F()
110 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
[all …]
Dconstant_folding_test.cc44 Output x = ops::Placeholder(s.WithOpName("x"), DTYPE, in SimpleNeutralElementTest()
46 Output v = ops::Variable(s.WithOpName("v"), {2, 2}, DTYPE); in SimpleNeutralElementTest()
55 Output zeros = ops::Const(s.WithOpName("zeros"), zeros_t); in SimpleNeutralElementTest()
56 Output ones = ops::Const(s.WithOpName("ones"), ones_t); in SimpleNeutralElementTest()
62 mul1 = ops::LogicalAnd(s.WithOpName("mul1"), x, zeros); in SimpleNeutralElementTest()
63 mul2 = ops::LogicalAnd(s.WithOpName("mul2"), x, ones); in SimpleNeutralElementTest()
64 add1 = ops::LogicalOr(s.WithOpName("add1"), x, zeros); in SimpleNeutralElementTest()
65 add2 = ops::LogicalOr(s.WithOpName("add2"), x, ones); in SimpleNeutralElementTest()
68 mul1 = ops::MulNoNan(s.WithOpName("mul1"), x, zeros); in SimpleNeutralElementTest()
70 mul1 = ops::Mul(s.WithOpName("mul1"), x, zeros); in SimpleNeutralElementTest()
[all …]
Dauto_mixed_precision_test.cc169 Output eye = ops::Const(s.WithOpName("eye"), in TestSimpleUnaryInferOp()
171 Output input = ops::Placeholder(s.WithOpName("input"), DT_FLOAT); in TestSimpleUnaryInferOp()
172 Output allow1 = ops::MatMul(s.WithOpName("allow1"), input, eye); in TestSimpleUnaryInferOp()
173 Output infer1 = test_op_factory(s.WithOpName("infer1"), allow1); in TestSimpleUnaryInferOp()
174 Output allow2 = ops::MatMul(s.WithOpName("allow2"), infer1, eye); in TestSimpleUnaryInferOp()
175 Output fetch1 = ops::Identity(s.WithOpName("fetch1"), allow2); in TestSimpleUnaryInferOp()
211 Output input = ops::Const(s.WithOpName("input"), 1.234f, {32}); in TEST_F()
212 Output deny1 = ops::Exp(s.WithOpName("deny1"), input); in TEST_F()
213 Output clr1 = ops::Relu(s.WithOpName("clr1"), deny1); in TEST_F()
214 Output infer1 = ops::Sqrt(s.WithOpName("infer1"), clr1); in TEST_F()
[all …]
Dmemory_optimizer_test.cc45 Output a = ops::Variable(s.WithOpName("a"), {2, 3, 4}, DT_FLOAT); in TEST_F()
46 Output b = ops::Identity(s.WithOpName("b"), a); // Recomputed in TEST_F()
47 Output c = ops::Identity(s.WithOpName("c"), b); in TEST_F()
48 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()
83 Output a = ops::Variable(s.WithOpName("a"), {2, 3, 4}, DT_FLOAT); in TEST_F()
84 Output b = ops::Identity(s.WithOpName("b"), a); // Would be recomputed, but in TEST_F()
86 Output c = ops::Identity(s.WithOpName("c"), b); in TEST_F()
87 Output d = ops::AddN(s.WithOpName("gradients/d"), {c}); in TEST_F()
[all …]
Dlayout_optimizer_test.cc80 ops::Const(s->WithOpName("Input"), Input::Initializer(input_data)); in SimpleConv2D()
87 ops::Const(s->WithOpName("Filter"), Input::Initializer(filter_data)); in SimpleConv2D()
95 Output conv = ops::Conv2D(s->WithOpName("Conv2D").WithDevice(device), input, in SimpleConv2D()
142 ops::Const(s->WithOpName("InputSizes"), Input::Initializer(input_data)); in SimpleConv2DBackpropInput()
147 ops::Variable(s->WithOpName("Filter"), filter_shape, DT_FLOAT); in SimpleConv2DBackpropInput()
154 ops::Const(s->WithOpName("Output"), Input::Initializer(output_data)); in SimpleConv2DBackpropInput()
158 ops::Identity(s->WithOpName("InputSizesIdentity"), input_sizes); in SimpleConv2DBackpropInput()
176 s->WithOpName("Conv2DBackpropInput"), input_sizes, filter, output, in SimpleConv2DBackpropInput()
180 s->WithOpName("Conv2DBackpropInput"), input_sizes_i, filter, output, in SimpleConv2DBackpropInput()
204 Output x = ops::Const(s->WithOpName("Input"), Input::Initializer(data)); in SimpleFusedBatchNormGrad()
[all …]
Dremapper_test.cc46 Output dflt = ops::Const(s.WithOpName("dflt"), {3.14f, 2.7f}, {2, 1, 1, 1}); in TEST_F()
47 Output x = ops::PlaceholderWithDefault(s.WithOpName("x"), dflt, {2, 1, 1, 1}); in TEST_F()
48 Output scale = ops::Const(s.WithOpName("scale"), {0.3f}, {1}); in TEST_F()
49 Output offset = ops::Const(s.WithOpName("offset"), {0.123f}, {1}); in TEST_F()
50 Output mean = ops::Const(s.WithOpName("mean"), {7.3f}, {1}); in TEST_F()
51 Output variance = ops::Const(s.WithOpName("variance"), {0.57f}, {1}); in TEST_F()
54 ops::FusedBatchNorm bn(s.WithOpName("batch_norm"), x, scale, offset, mean, in TEST_F()
79 ops::Const(s.WithOpName("dflt"), {3.14f, 2.7f, 1.0f, 2.0f, 3.0f, 100.0f}, in TEST_F()
81 Output x = ops::PlaceholderWithDefault(s.WithOpName("x"), dflt, {1, 3, 1, 2}); in TEST_F()
82 Output scale = ops::Const(s.WithOpName("scale"), {0.3f, 7.0f, 123.0f}, {3}); in TEST_F()
[all …]
Dscoped_allocator_optimizer_test.cc88 ops::Const<float>(s.WithOpName("a"), {1.0, 0.0, 0.0, -1.0}, {2, 2}); in BuildAbsGraph()
90 ops::Const<float>(s.WithOpName("b"), {1.0, -2.0, 3.0, 4.0}, {2, 2}); in BuildAbsGraph()
92 ops::Const<float>(s.WithOpName("c"), {-5.0, -2.0, 0.0, -2.0}, {2, 2}); in BuildAbsGraph()
93 Output s1 = ops::Add(s.WithOpName("s1"), a, b); in BuildAbsGraph()
94 Output s2 = ops::Add(s.WithOpName("s2"), b, c); in BuildAbsGraph()
97 int1 = ops::Identity(s.WithOpName("i1"), s1); in BuildAbsGraph()
98 int2 = ops::Identity(s.WithOpName("i2"), s2); in BuildAbsGraph()
103 Output a1 = ops::Abs(s.WithOpName("a1"), int1); in BuildAbsGraph()
104 Output a2 = ops::Abs(s.WithOpName("a2"), int2); in BuildAbsGraph()
105 Output r1 = ops::Reshape(s.WithOpName("r1"), a1, {1, 4}); in BuildAbsGraph()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass_test.cc41 auto a = ops::Placeholder(scope.WithOpName("A"), DT_INT32); in MakeOuterGraph()
42 auto b = ops::Placeholder(scope.WithOpName("B"), DT_FLOAT); in MakeOuterGraph()
43 auto c = ops::Placeholder(scope.WithOpName("C"), DT_INT32); in MakeOuterGraph()
44 auto d = ops::Placeholder(scope.WithOpName("D"), DT_FLOAT); in MakeOuterGraph()
45 auto u = ops::Placeholder(scope.WithOpName("U"), DT_RESOURCE); in MakeOuterGraph()
46 auto v = ops::Placeholder(scope.WithOpName("V"), DT_RESOURCE); in MakeOuterGraph()
47 auto w = ops::Placeholder(scope.WithOpName("W"), DT_RESOURCE); in MakeOuterGraph()
76 ops::XlaClusterOutput(scope.WithOpName("Out0"), Output(launch, 0)); in MakeOuterGraph()
78 ops::XlaClusterOutput(scope.WithOpName("Out1"), Output(launch, 1)); in MakeOuterGraph()
80 ops::XlaClusterOutput(scope.WithOpName("Out2"), Output(launch, 2)); in MakeOuterGraph()
[all …]
Dshape_inference_test.cc38 auto a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT, in TEST()
40 auto b = ops::Placeholder(root.WithOpName("B"), DT_FLOAT, in TEST()
42 auto c = ops::Placeholder(root.WithOpName("C"), DT_FLOAT); in TEST()
43 auto d = ops::Add(root.WithOpName("D"), a, b); in TEST()
44 auto e = ops::Add(root.WithOpName("E"), d, c); in TEST()
45 auto f = ops::Neg(root.WithOpName("F"), e); in TEST()
46 auto g = ops::AddN(root.WithOpName("G"), std::initializer_list<Output>{e, f}); in TEST()
72 auto dummy = ops::Placeholder(scope.WithOpName("Dummy"), DT_INT32, in TEST()
75 auto source = ops::Placeholder(scope.WithOpName("source"), DT_INT32, in TEST()
78 ops::internal::Enter(scope.WithOpName("while/Enter"), source, "aloop"); in TEST()
[all …]
Ddeadness_analysis_test.cc65 Output value = ops::Placeholder(root.WithOpName(prefix + "/value"), DT_FLOAT); in CreateSwitch()
67 ops::Placeholder(root.WithOpName(prefix + "/pred"), DT_BOOL); in CreateSwitch()
68 return ops::Switch(root.WithOpName(prefix + "/switch"), value, predicate); in CreateSwitch()
134 root.WithOpName(prefix + "/enter"), initial_value, frame_name); in CreateInductionVariable()
136 ops::Merge iv(root.WithOpName(prefix + "/iv"), in CreateInductionVariable()
138 Output increment_by = ops::Const(root.WithOpName(prefix + "/incr"), 1); in CreateInductionVariable()
139 Output final_value = ops::Const(root.WithOpName(prefix + "/final"), 10); in CreateInductionVariable()
141 ops::Less(root.WithOpName(prefix + "/cond"), iv.output, final_value); in CreateInductionVariable()
142 ops::Switch latch(root.WithOpName(prefix + "/latch"), iv.output, in CreateInductionVariable()
144 ops::internal::Exit exit(root.WithOpName(prefix + "/exit"), in CreateInductionVariable()
[all …]
Dxla_cluster_util_test.cc43 Output a = ops::Const(root.WithOpName("a"), Input::Initializer(0.0)); in TEST()
45 ops::internal::Enter(root.WithOpName("enter"), a, "only_frame"); in TEST()
46 Output exit = ops::internal::Exit(root.WithOpName("exit"), enter); in TEST()
47 Output b = ops::Add(root.WithOpName("b"), a, exit); in TEST()
59 Output a = ops::Const(root.WithOpName("a"), Input::Initializer(0.0)); in TEST()
61 ops::internal::Enter(root.WithOpName("enter_0"), a, "frame_0"); in TEST()
62 Output exit_0 = ops::internal::Exit(root.WithOpName("exit_0"), enter_0); in TEST()
64 ops::internal::Enter(root.WithOpName("enter_1"), a, "frame_1"); in TEST()
65 Output exit_1 = ops::internal::Exit(root.WithOpName("exit_1"), enter_1); in TEST()
66 Output b = ops::Add(root.WithOpName("b"), a, exit_1); in TEST()
[all …]
Dincrease_dynamism_for_auto_jit_pass_test.cc106 Output input = ops::Placeholder(root.WithOpName("input"), DT_FLOAT); in TEST()
107 Output begin = ops::Placeholder(root.WithOpName("begin"), DT_INT32); in TEST()
108 Output size = ops::Const(root.WithOpName("size"), {-1, 500}); in TEST()
109 Output slice = ops::Slice(root.WithOpName("slice"), input, begin, size); in TEST()
152 Output input = ops::Placeholder(root.WithOpName("input"), DT_FLOAT); in TEST()
153 Output begin = ops::Placeholder(root.WithOpName("begin"), DT_INT32); in TEST()
154 Output size = ops::Const(root.WithOpName("size"), {-1}); in TEST()
155 Output slice = ops::Slice(root.WithOpName("slice"), input, begin, size); in TEST()
172 Output input = ops::Placeholder(root.WithOpName("input"), DT_FLOAT); in TEST()
173 Output begin = ops::Placeholder(root.WithOpName("begin"), DT_INT32); in TEST()
[all …]
Dmark_for_compilation_pass_test.cc215 Output a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT); in TEST()
219 ops::PartitionedCall b(root.WithOpName("B"), {a, a}, {DT_FLOAT}, b_name_attr); in TEST()
223 ops::PartitionedCall c(root.WithOpName("C"), {a}, {DT_FLOAT}, c_name_attr); in TEST()
224 Output d = ops::Add(root.WithOpName("D"), b.output.front(), c.output.front()); in TEST()
427 auto a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT); in TEST()
428 auto b = ops::Placeholder(root.WithOpName("B"), DT_FLOAT); in TEST()
429 auto c = ops::Add(root.WithOpName("C"), a, b); in TEST()
433 auto d = ops::Add(root.WithOpName("D"), c, exit); in TEST()
583 Output cond_a = ops::Placeholder(root.WithOpName("cond_a"), DT_BOOL); in TEST()
584 Output cond_b = ops::Placeholder(root.WithOpName("cond_b"), DT_BOOL); in TEST()
[all …]
Dclone_constants_for_better_clustering_test.cc69 Output in0 = ops::Placeholder(on_gpu.WithOpName("in0"), DT_FLOAT); in TEST()
70 Output in1 = ops::Placeholder(on_gpu.WithOpName("in1"), DT_FLOAT); in TEST()
72 Output perm = ops::Const(on_cpu.WithOpName("perm"), {3, 1, 2, 0}); in TEST()
75 Output tr0 = ops::Transpose(on_gpu.WithOpName("tr0"), in0, perm); in TEST()
76 Output tr1 = ops::Transpose(on_gpu.WithOpName("tr1"), in1, perm); in TEST()
95 Output in0 = ops::Placeholder(on_gpu.WithOpName("in0"), DT_FLOAT); in TEST()
96 Output in1 = ops::Placeholder(on_gpu.WithOpName("in1"), DT_FLOAT); in TEST()
98 Output perm = ops::Const(on_gpu.WithOpName("perm"), {3, 1, 2, 0}); in TEST()
101 Output tr0 = ops::Transpose(on_gpu.WithOpName("tr0"), in0, perm); in TEST()
102 Output tr1 = ops::Transpose(on_gpu.WithOpName("tr1"), in1, perm); in TEST()
[all …]
Drearrange_function_argument_pass_test.cc48 Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_RESOURCE, 0); in TEST()
49 Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_BOOL, 1); in TEST()
50 auto ret0 = ops::_Retval(s.WithOpName("ret0"), arg1, 0); in TEST()
51 auto ret1 = ops::_Retval(s.WithOpName("ret1"), arg0, 1); in TEST()
63 Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_RESOURCE, 0); in TEST()
64 Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_BOOL, 1); in TEST()
65 auto ret0 = ops::_Retval(s.WithOpName("ret0"), arg0, 0); in TEST()
66 auto ret1 = ops::_Retval(s.WithOpName("ret1"), arg1, 1); in TEST()
77 Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_RESOURCE, 0); in TEST()
78 Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_BOOL, 1); in TEST()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc164 return Const(root.WithOpName(name), Input::Initializer(tensor)); in TestIgnoreOps()
175 Output mat_mul_op = MatMul(root.WithOpName("mat_mul_op"), a_op, b_op); in TestIgnoreOps()
176 Output mul_op = Mul(root.WithOpName("mul"), c_op, d_op); in TestIgnoreOps()
215 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); in TestQuantizeMatMul()
219 Output b_op = Const(root.WithOpName("b_op"), Input::Initializer(b_tensor)); in TestQuantizeMatMul()
221 Output mat_mul_op = MatMul(root.WithOpName("mat_mul_op"), a_op, b_op); in TestQuantizeMatMul()
266 Output x = Const(root.WithOpName("x"), Input::Initializer(x_float_tensor)); in TestQuantizeMul()
270 Output y = Const(root.WithOpName("y"), Input::Initializer(y_float_tensor)); in TestQuantizeMul()
272 Mul mul = Mul(root.WithOpName("mul"), x, y); in TestQuantizeMul()
301 Output x = Const(root.WithOpName("x"), Input::Initializer(x_float_tensor)); in TestQuantizeAdd()
[all …]
Dfold_constants_test.cc55 Const(root.WithOpName("a_expect_removed"), Input::Initializer(a_data)); in TestSimpleAdd()
60 Const(root.WithOpName("b_expect_removed"), Input::Initializer(b_data)); in TestSimpleAdd()
62 Output add = Add(root.WithOpName("add_expect_removed"), a_const, b_const); in TestSimpleAdd()
65 Placeholder(root.WithOpName("placeholder_expect_remains"), DT_FLOAT); in TestSimpleAdd()
68 Mul(root.WithOpName("output_expect_remains"), add, placeholder); in TestSimpleAdd()
92 Const(root.WithOpName("a_expect_remains"), Input::Initializer(a_data)); in TestOpExclusionAdd()
97 Const(root.WithOpName("b_expect_remains"), Input::Initializer(b_data)); in TestOpExclusionAdd()
99 Output add = Add(root.WithOpName("add_expect_remains"), a_const, b_const); in TestOpExclusionAdd()
102 Placeholder(root.WithOpName("placeholder_expect_remains"), DT_FLOAT); in TestOpExclusionAdd()
105 Mul(root.WithOpName("output_expect_remains"), add, placeholder); in TestOpExclusionAdd()
[all …]
Dfold_old_batch_norms_test.cc49 Const(root.WithOpName("input_op"), Input::Initializer(input_data)); in TestFoldOldBatchNorms()
55 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldOldBatchNorms()
57 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldOldBatchNorms()
63 Const(root.WithOpName("mean_op"), Input::Initializer(mean_data)); in TestFoldOldBatchNorms()
67 Output variance_op = Const(root.WithOpName("variance_op"), in TestFoldOldBatchNorms()
73 Const(root.WithOpName("beta_op"), Input::Initializer(beta_data)); in TestFoldOldBatchNorms()
78 Const(root.WithOpName("gamma_op"), Input::Initializer(gamma_data)); in TestFoldOldBatchNorms()
133 Const(root.WithOpName("input_op"), Input::Initializer(input_data)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
139 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
141 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc824 auto arg = ops::_Arg(s.WithOpName("x"), DT_FLOAT, 0); in TEST_F()
827 auto ret = ops::_Retval(s.WithOpName("y_RetVal"), b, 0); in TEST_F()
840 auto x = ops::_Arg(s.WithOpName("x"), DT_FLOAT, 0); in TEST_F()
841 auto func0 = ops::Identity(s.WithOpName("Func/x4/input/_0"), x); in TEST_F()
844 auto func1 = ops::Identity(s.WithOpName("Func/x4/output/_1"), x4_y); in TEST_F()
845 auto func2 = ops::Identity(s.WithOpName("Func/y/input/_2"), func1); in TEST_F()
848 auto func3 = ops::Identity(s.WithOpName("Func/y/output/_3"), y_y); in TEST_F()
849 auto ret = ops::_Retval(s.WithOpName("y_RetVal"), func3, 0); in TEST_F()
862 auto x = ops::_Arg(s.WithOpName("x"), DT_FLOAT, 0); in TEST_F()
863 auto x4_x2_two = ops::Const<int64>(s.WithOpName("x4/x2/two"), int64{2}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment_test.cc146 auto feed = ops::Placeholder(s.WithOpName("feed"), DT_FLOAT); in TEST_F()
147 auto add0 = ops::Add(s.WithOpName("add0"), feed, feed); in TEST_F()
148 auto add1 = ops::Add(s.WithOpName("add1"), feed, feed); in TEST_F()
149 auto add2 = ops::Add(s.WithOpName("add2"), add0, add1); in TEST_F()
150 auto add3 = ops::Add(s.WithOpName("add3"), add0, add2); in TEST_F()
151 auto add4 = ops::Add(s.WithOpName("add4"), add2, add2); in TEST_F()
194 auto feed = ops::Placeholder(s.WithOpName("feed"), DT_FLOAT); in TEST_F()
195 auto add0 = ops::Add(s.WithOpName("add0"), feed, feed); in TEST_F()
196 auto add1 = ops::Add(s.WithOpName("add1"), feed, feed); in TEST_F()
197 auto add2 = ops::Add(s.WithOpName("add2"), add0, add1); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc241 auto a1 = FloatInput(in_.WithOpName("A1")); in TEST_F()
242 Combine(in_.WithOpName("A2"), a1, a1); in TEST_F()
247 a1 = FloatInput(scope_a_.WithOpName("A1")); in TEST_F()
248 Combine(scope_a_.WithOpName("A2"), a1, a1); in TEST_F()
253 auto a1 = FloatInput(in_.WithOpName("A1")); in TEST_F()
254 auto b1 = FloatInput(in_.WithOpName("B1")); in TEST_F()
255 Combine(in_.WithOpName("B2"), a1, b1); in TEST_F()
262 a1 = FloatInput(scope_a_.WithOpName("A1")); in TEST_F()
263 _Send(scope_a_.WithOpName("A1/_0"), a1, "edge_1_A1", a, 82, b); in TEST_F()
266 b1 = FloatInput(scope_b_.WithOpName("B1")); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test_test.cc31 auto s1_a = ops::Variable(s1.WithOpName("a"), {2, 2}, DT_FLOAT); in TEST_F()
32 auto s1_b = ops::Variable(s1.WithOpName("b"), {2, 2}, DT_FLOAT); in TEST_F()
33 auto s1_add = ops::Add(s1.WithOpName("Add_1"), s1_a, s1_b); in TEST_F()
36 auto s2_a = ops::Variable(s2.WithOpName("a"), {2, 2}, DT_FLOAT); in TEST_F()
37 auto s2_b = ops::Variable(s2.WithOpName("b"), {2, 2}, DT_FLOAT); in TEST_F()
38 auto s2_add = ops::Add(s2.WithOpName("Add_1"), s2_a, s2_b); in TEST_F()
52 auto a = ops::Variable(s.WithOpName("a"), {2, 2}, DT_FLOAT); in TEST_F()
53 auto b = ops::Variable(s.WithOpName("b"), {2, 2}, DT_FLOAT); in TEST_F()
54 auto add_1 = ops::Add(s.WithOpName("Add_1"), a, b); in TEST_F()
55 auto add_2 = ops::Add(s.WithOpName("Add_2"), add_1, b); in TEST_F()
[all …]

123456