/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | functionalize_control_flow_test.cc | 107 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/ |
D | model_pruner_test.cc | 59 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 …]
|
D | arithmetic_optimizer_test.cc | 111 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 …]
|
D | dependency_optimizer_test.cc | 69 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 …]
|
D | constant_folding_test.cc | 44 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 …]
|
D | auto_mixed_precision_test.cc | 169 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 …]
|
D | memory_optimizer_test.cc | 45 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 …]
|
D | layout_optimizer_test.cc | 80 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 …]
|
D | remapper_test.cc | 46 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 …]
|
D | scoped_allocator_optimizer_test.cc | 88 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/ |
D | encapsulate_xla_computations_pass_test.cc | 41 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 …]
|
D | shape_inference_test.cc | 38 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 …]
|
D | deadness_analysis_test.cc | 65 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 …]
|
D | xla_cluster_util_test.cc | 43 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 …]
|
D | increase_dynamism_for_auto_jit_pass_test.cc | 106 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 …]
|
D | mark_for_compilation_pass_test.cc | 215 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 …]
|
D | clone_constants_for_better_clustering_test.cc | 69 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 …]
|
D | rearrange_function_argument_pass_test.cc | 48 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/ |
D | quantize_nodes_test.cc | 164 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 …]
|
D | fold_constants_test.cc | 55 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 …]
|
D | fold_old_batch_norms_test.cc | 49 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/ |
D | function_test.cc | 824 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/ |
D | segment_test.cc | 146 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/ |
D | graph_partition_test.cc | 241 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/ |
D | grappler_test_test.cc | 31 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 …]
|