Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc70 auto x = ops::Placeholder(scope.WithOpName("x"), DT_INT32); in TEST()
71 auto y = ops::Placeholder(scope.WithOpName("y"), DT_INT32); in TEST()
72 auto less = ops::Less(scope.WithOpName("cond/Less"), y, x); in TEST()
73 auto switch_1 = ops::Switch(scope.WithOpName("cond/Switch"), less, less); in TEST()
76 ops::Identity(scope.WithOpName("cond/Identity"), switch_1.output_true); in TEST()
78 scope.WithOpName("cond").WithControlDependencies(identity_t), 17); in TEST()
79 auto switch_2 = ops::Switch(scope.WithOpName("cond/Switch"), y, less); in TEST()
80 auto mul = ops::Multiply(scope.WithOpName("cond/Mul"), switch_2.output_true, in TEST()
84 ops::Identity(scope.WithOpName("cond/Identity"), switch_1.output_false); in TEST()
86 scope.WithOpName("cond").WithControlDependencies(identity_f), 23); in TEST()
[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 …]
Dconstant_folding_test.cc43 Output x = ops::Placeholder(s.WithOpName("x"), DTYPE, in SimpleNeutralElementTest()
45 Output v = ops::Variable(s.WithOpName("v"), {2, 2}, DTYPE); in SimpleNeutralElementTest()
54 Output zeros = ops::Const(s.WithOpName("zeros"), zeros_t); in SimpleNeutralElementTest()
55 Output ones = ops::Const(s.WithOpName("ones"), ones_t); in SimpleNeutralElementTest()
61 mul1 = ops::LogicalAnd(s.WithOpName("mul1"), x, zeros); in SimpleNeutralElementTest()
62 mul2 = ops::LogicalAnd(s.WithOpName("mul2"), x, ones); in SimpleNeutralElementTest()
63 add1 = ops::LogicalOr(s.WithOpName("add1"), x, zeros); in SimpleNeutralElementTest()
64 add2 = ops::LogicalOr(s.WithOpName("add2"), x, ones); in SimpleNeutralElementTest()
67 mul1 = ops::MulNoNan(s.WithOpName("mul1"), x, zeros); in SimpleNeutralElementTest()
69 mul1 = ops::Mul(s.WithOpName("mul1"), x, zeros); in SimpleNeutralElementTest()
[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 …]
Darithmetic_optimizer_test.cc108 Output c1 = ops::Const(s.WithOpName("c1"), {3.14, 2.7}, {1, 2}); in TEST_F()
109 Output c2 = ops::Const(s.WithOpName("c2"), {3.14, 2.7}, {1, 2}); in TEST_F()
110 Output div = ops::Div(s.WithOpName("div"), c1, c2); in TEST_F()
140 Output c = ops::Const(s.WithOpName("c"), {3.14, 2.7}, {1, 2}); in TEST_F()
141 auto check1 = ops::CheckNumerics(s.WithOpName("check1"), c, "foo"); in TEST_F()
142 auto check2 = ops::CheckNumerics(s.WithOpName("check2"), c, "foo"); in TEST_F()
143 auto assert1 = ops::Assert(s.WithOpName("assert1"), p, {c}); in TEST_F()
144 auto assert2 = ops::Assert(s.WithOpName("assert2"), p, {c}); in TEST_F()
145 Output div = ops::Div(s.WithOpName("div").WithControlDependencies( in TEST_F()
178 Output c1 = ops::Const(s.WithOpName("c1"), {1.0f, 2.0f}, {1, 2}); 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 …]
Dauto_mixed_precision_test.cc142 Output eye = ops::Const(s.WithOpName("eye"), in TestSimpleUnaryGrayOp()
144 Output input = ops::Placeholder(s.WithOpName("input"), DT_FLOAT); in TestSimpleUnaryGrayOp()
145 Output wht1 = ops::MatMul(s.WithOpName("wht1"), input, eye); in TestSimpleUnaryGrayOp()
146 Output gry1 = test_op_factory(s.WithOpName("gry1"), wht1); in TestSimpleUnaryGrayOp()
147 Output wht2 = ops::MatMul(s.WithOpName("wht2"), gry1, eye); in TestSimpleUnaryGrayOp()
148 Output fetch1 = ops::Identity(s.WithOpName("fetch1"), wht2); in TestSimpleUnaryGrayOp()
203 Output input = ops::Const(s.WithOpName("input"), 1.234f, {32}); in TEST_F()
204 Output blk1 = ops::Exp(s.WithOpName("blk1"), input); in TEST_F()
205 Output clr1 = ops::Relu(s.WithOpName("clr1"), blk1); in TEST_F()
206 Output gry1 = ops::Sqrt(s.WithOpName("gry1"), clr1); in TEST_F()
[all …]
Dlayout_optimizer_test.cc75 ops::Const(s->WithOpName("Input"), Input::Initializer(input_data)); in SimpleConv2D()
82 ops::Const(s->WithOpName("Filter"), Input::Initializer(filter_data)); in SimpleConv2D()
89 Output conv = ops::Conv2D(s->WithOpName("Conv2D").WithDevice(device), input, in SimpleConv2D()
136 ops::Const(s->WithOpName("InputSizes"), Input::Initializer(input_data)); in SimpleConv2DBackpropInput()
141 ops::Variable(s->WithOpName("Filter"), filter_shape, DT_FLOAT); in SimpleConv2DBackpropInput()
148 ops::Const(s->WithOpName("Output"), Input::Initializer(output_data)); in SimpleConv2DBackpropInput()
152 ops::Identity(s->WithOpName("InputSizesIdentity"), input_sizes); in SimpleConv2DBackpropInput()
170 s->WithOpName("Conv2DBackpropInput"), input_sizes, filter, output, in SimpleConv2DBackpropInput()
174 s->WithOpName("Conv2DBackpropInput"), input_sizes_i, filter, output, in SimpleConv2DBackpropInput()
198 Output x = ops::Const(s->WithOpName("Input"), Input::Initializer(data)); in SimpleFusedBatchNormGrad()
[all …]
Dremapper_test.cc44 Output dflt = ops::Const(s.WithOpName("dflt"), {3.14f, 2.7f}, {2, 1, 1, 1}); in TEST_F()
45 Output x = ops::PlaceholderWithDefault(s.WithOpName("x"), dflt, {2, 1, 1, 1}); in TEST_F()
46 Output scale = ops::Const(s.WithOpName("scale"), {0.3f}, {1}); in TEST_F()
47 Output offset = ops::Const(s.WithOpName("offset"), {0.123f}, {1}); in TEST_F()
48 Output mean = ops::Const(s.WithOpName("mean"), {7.3f}, {1}); in TEST_F()
49 Output variance = ops::Const(s.WithOpName("variance"), {0.57f}, {1}); in TEST_F()
52 ops::FusedBatchNorm bn(s.WithOpName("batch_norm"), x, scale, offset, mean, in TEST_F()
77 ops::Const(s.WithOpName("dflt"), {3.14f, 2.7f, 1.0f, 2.0f, 3.0f, 100.0f}, in TEST_F()
79 Output x = ops::PlaceholderWithDefault(s.WithOpName("x"), dflt, {1, 3, 1, 2}); in TEST_F()
80 Output scale = ops::Const(s.WithOpName("scale"), {0.3f, 7.0f, 123.0f}, {3}); in TEST_F()
[all …]
Dscoped_allocator_optimizer_test.cc86 ops::Const<float>(s.WithOpName("a"), {1.0, 0.0, 0.0, -1.0}, {2, 2}); in BuildAbsGraph()
88 ops::Const<float>(s.WithOpName("b"), {1.0, -2.0, 3.0, 4.0}, {2, 2}); in BuildAbsGraph()
90 ops::Const<float>(s.WithOpName("c"), {-5.0, -2.0, 0.0, -2.0}, {2, 2}); in BuildAbsGraph()
91 Output s1 = ops::Add(s.WithOpName("s1"), a, b); in BuildAbsGraph()
92 Output s2 = ops::Add(s.WithOpName("s2"), b, c); in BuildAbsGraph()
95 int1 = ops::Identity(s.WithOpName("i1"), s1); in BuildAbsGraph()
96 int2 = ops::Identity(s.WithOpName("i2"), s2); in BuildAbsGraph()
101 Output a1 = ops::Abs(s.WithOpName("a1"), int1); in BuildAbsGraph()
102 Output a2 = ops::Abs(s.WithOpName("a2"), int2); in BuildAbsGraph()
103 Output r1 = ops::Reshape(s.WithOpName("r1"), a1, {1, 4}); in BuildAbsGraph()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass_test.cc40 auto a = ops::Placeholder(scope.WithOpName("A"), DT_INT32); in MakeOuterGraph()
41 auto b = ops::Placeholder(scope.WithOpName("B"), DT_FLOAT); in MakeOuterGraph()
42 auto c = ops::Placeholder(scope.WithOpName("C"), DT_INT32); in MakeOuterGraph()
43 auto d = ops::Placeholder(scope.WithOpName("D"), DT_FLOAT); in MakeOuterGraph()
44 auto u = ops::Placeholder(scope.WithOpName("U"), DT_RESOURCE); in MakeOuterGraph()
45 auto v = ops::Placeholder(scope.WithOpName("V"), DT_RESOURCE); in MakeOuterGraph()
46 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.cc66 Output value = ops::Placeholder(root.WithOpName(prefix + "/value"), DT_FLOAT); in CreateSwitch()
68 ops::Placeholder(root.WithOpName(prefix + "/pred"), DT_BOOL); in CreateSwitch()
69 return ops::Switch(root.WithOpName(prefix + "/switch"), value, predicate); in CreateSwitch()
135 root.WithOpName(prefix + "/enter"), initial_value, frame_name); in CreateInductionVariable()
137 ops::Merge iv(root.WithOpName(prefix + "/iv"), in CreateInductionVariable()
139 Output increment_by = ops::Const(root.WithOpName(prefix + "/incr"), 1); in CreateInductionVariable()
140 Output final_value = ops::Const(root.WithOpName(prefix + "/final"), 10); in CreateInductionVariable()
142 ops::Less(root.WithOpName(prefix + "/cond"), iv.output, final_value); in CreateInductionVariable()
143 ops::Switch latch(root.WithOpName(prefix + "/latch"), iv.output, in CreateInductionVariable()
145 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 …]
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 …]
Dmark_for_compilation_pass_test.cc211 Output a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT); in TEST()
215 ops::PartitionedCall b(root.WithOpName("B"), {a, a}, {DT_FLOAT}, b_name_attr); in TEST()
219 ops::PartitionedCall c(root.WithOpName("C"), {a}, {DT_FLOAT}, c_name_attr); in TEST()
220 Output d = ops::Add(root.WithOpName("D"), b.output.front(), c.output.front()); in TEST()
423 auto a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT); in TEST()
424 auto b = ops::Placeholder(root.WithOpName("B"), DT_FLOAT); in TEST()
425 auto c = ops::Add(root.WithOpName("C"), a, b); in TEST()
429 auto d = ops::Add(root.WithOpName("D"), c, exit); in TEST()
579 Output cond_a = ops::Placeholder(root.WithOpName("cond_a"), DT_BOOL); in TEST()
580 Output cond_b = ops::Placeholder(root.WithOpName("cond_b"), DT_BOOL); 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/core/common_runtime/
Dfunction_test.cc697 auto arg = ops::_Arg(s.WithOpName("x"), DT_FLOAT, 0); in TEST_F()
700 auto ret = ops::_Retval(s.WithOpName("y_RetVal"), b, 0); in TEST_F()
713 auto x = ops::_Arg(s.WithOpName("x"), DT_FLOAT, 0); in TEST_F()
714 auto func0 = ops::Identity(s.WithOpName("Func/x4/input/_0"), x); in TEST_F()
717 auto func1 = ops::Identity(s.WithOpName("Func/x4/output/_1"), x4_y); in TEST_F()
718 auto func2 = ops::Identity(s.WithOpName("Func/y/input/_2"), func1); in TEST_F()
721 auto func3 = ops::Identity(s.WithOpName("Func/y/output/_3"), y_y); in TEST_F()
722 auto ret = ops::_Retval(s.WithOpName("y_RetVal"), func3, 0); in TEST_F()
735 auto x = ops::_Arg(s.WithOpName("x"), DT_FLOAT, 0); in TEST_F()
736 auto x4_x2_two = ops::Const<int64>(s.WithOpName("x4/x2/two"), 2LL); in TEST_F()
[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/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/compiler/tf2tensorrt/segment/
Dsegment_test.cc124 auto feed = ops::Placeholder(s.WithOpName("feed"), DT_FLOAT); in TEST_F()
125 auto add0 = ops::Add(s.WithOpName("add0"), feed, feed); in TEST_F()
126 auto add1 = ops::Add(s.WithOpName("add1"), feed, feed); in TEST_F()
127 auto add2 = ops::Add(s.WithOpName("add2"), add0, add1); in TEST_F()
128 auto add3 = ops::Add(s.WithOpName("add3"), add0, add2); in TEST_F()
129 auto add4 = ops::Add(s.WithOpName("add4"), add2, add2); in TEST_F()
171 auto feed = ops::Placeholder(s.WithOpName("feed"), DT_FLOAT); in TEST_F()
172 auto add0 = ops::Add(s.WithOpName("add0"), feed, feed); in TEST_F()
173 auto add1 = ops::Add(s.WithOpName("add1"), feed, feed); in TEST_F()
174 auto add2 = ops::Add(s.WithOpName("add2"), add0, add1); 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