| /external/tensorflow/tensorflow/core/framework/ |
| node_def_builder_test.cc | 118 ExpectSuccess(Builder().Input("x", 0, DT_INT32), {DT_INT32}, {DT_FLOAT},
122 ExpectSuccess(Builder().Input("y", 2, DT_INT32), {DT_INT32}, {DT_FLOAT},
126 ExpectSuccess(Builder().Input(FakeInput()), {DT_INT32}, {DT_FLOAT}, R"proto(
129 ExpectSuccess(Builder().Input(FakeInput(DT_INT32)), {DT_INT32}, {DT_FLOAT},
133 ExpectSuccess(Builder().Input(FakeInput(DT_INT32_REF)), {DT_INT32},
139 {DT_INT32}, {DT_FLOAT}, R"proto(
143 ExpectSuccess(Builder().Input(FakeInput()).Device("ddd"), {DT_INT32},
[all...] |
| function_testlib.cc | 111 {{"SrcT", DT_INT64}, {"DstT", DT_INT32}}},
112 {{"y"}, "Mul", {"x", "scale"}, {{"T", DT_INT32}}},
199 {{{"enter"}, "Enter", {"i"}, {{"T", DT_INT32}, {"frame_name", "while"}}},
200 {{"add"}, "Add", {"enter:output", "i"}, {{"T", DT_INT32}}}},
|
| /external/tensorflow/tensorflow/core/kernels/ |
| logging_ops_test.cc | 44 TF_ASSERT_OK(Init(DT_INT32, DT_INT32));
49 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
55 TF_ASSERT_OK(Init(DT_INT32, DT_INT32, "", -1, 6));
60 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
66 TF_ASSERT_OK(Init(DT_INT32, DT_STRING));
71 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
77 TF_ASSERT_OK(Init(DT_INT32, DT_STRING, "Message: "));
82 Tensor expected(allocator(), DT_INT32, TensorShape({6}))
[all...] |
| identity_n_op_test.cc | 40 TF_ASSERT_OK(Init(DT_INT32, DT_DOUBLE));
45 Tensor expected0(allocator(), DT_INT32, TensorShape({6}));
54 TF_ASSERT_OK(Init(DT_INT32, DT_INT32));
58 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3}));
66 TF_ASSERT_OK(Init(DT_STRING, DT_INT32));
73 Tensor expected1(allocator(), DT_INT32, TensorShape({8}));
|
| guarantee_const_op_test.cc | 41 TF_ASSERT_OK(Init(DT_INT32));
44 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
50 TF_ASSERT_OK(Init(DT_INT32));
53 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3}));
69 AddResourceInput("", "resource", new Var(DT_INT32));
|
| identity_op_test.cc | 40 TF_ASSERT_OK(Init(DT_INT32));
43 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
49 TF_ASSERT_OK(Init(DT_INT32));
52 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3}));
|
| control_flow_ops_test.cc | 40 Initialize(DT_INT32);
44 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
51 Initialize(DT_INT32);
55 Tensor expected(allocator(), DT_INT32, TensorShape({6}));
62 Initialize(DT_INT32);
66 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3}));
|
| roll_op_test.cc | 52 MakeOp(DT_FLOAT, DT_INT32);
67 MakeOp(DT_STRING, DT_INT32);
82 MakeOp(DT_COMPLEX64, DT_INT32);
103 MakeOp(DT_FLOAT, DT_INT32);
120 MakeOp(DT_STRING, DT_INT32);
138 MakeOp(DT_FLOAT, DT_INT32);
154 MakeOp(DT_STRING, DT_INT32);
241 MakeOp(DT_FLOAT, DT_INT32);
257 MakeOp(DT_STRING, DT_INT32);
275 MakeOp(DT_FLOAT, DT_INT32);
[all...] |
| bincount_op_test.cc | 30 Tensor arr(DT_INT32, TensorShape({arr_size}));
33 Tensor size(DT_INT32, TensorShape({static_cast<int32>(1)}));
36 Tensor weights(DT_INT32, TensorShape({0}));
43 .Attr("T", DT_INT32)
|
| sparse_to_dense_op_test.cc | 53 MakeOp(1, DT_INT32, DT_FLOAT);
91 MakeOp(1, DT_INT32, DT_FLOAT);
110 MakeOp(2, DT_INT32, DT_FLOAT);
132 MakeOp(2, DT_INT32, DT_FLOAT);
154 MakeOp(3, DT_INT32, DT_FLOAT);
176 MakeOp(3, DT_INT32, DT_FLOAT);
217 Tensor output_shape(DT_INT32, TensorShape({NDIM}));
218 Tensor sparse_indices(DT_INT32, TensorShape({N, NDIM}));
239 .Input(FakeInput(DT_INT32))
240 .Input(FakeInput(DT_INT32))
[all...] |
| unique_op_test.cc | 40 tensor_proto.set_dtype(DT_INT32);
53 tensor_proto.set_dtype(DT_INT32);
69 Tensor input(DT_INT32, TensorShape({dim}));
75 .Attr("T", DT_INT32)
88 Tensor input(DT_INT32, TensorShape({dim * 200}));
95 .Attr("T", DT_INT32)
|
| cross_op_test.cc | 82 .Input(FakeInput(DT_INT32))
83 .Input(FakeInput(DT_INT32))
94 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3}));
|
| non_max_suppression_op_test.cc | 37 .Input(FakeInput(DT_INT32))
54 Tensor expected(allocator(), DT_INT32, TensorShape({3}));
68 Tensor expected(allocator(), DT_INT32, TensorShape({3}));
83 Tensor expected(allocator(), DT_INT32, TensorShape({2}));
98 Tensor expected(allocator(), DT_INT32, TensorShape({3}));
110 Tensor expected(allocator(), DT_INT32, TensorShape({1}));
133 Tensor expected(allocator(), DT_INT32, TensorShape({1}));
174 Tensor expected(allocator(), DT_INT32, TensorShape({0}));
189 .Input(FakeInput(DT_INT32))
207 Tensor expected(allocator(), DT_INT32, TensorShape({3}))
[all...] |
| sequence_ops_test.cc | 56 MakeOp(DT_INT32);
65 Tensor expected(allocator(), DT_INT32, TensorShape({5}));
103 MakeOp(DT_FLOAT, DT_INT32);
133 MakeOp(DT_DOUBLE, DT_INT32);
|
| diag_op_test.cc | 42 BM_DiagDev(N, int, DT_INT32, cpu); \
45 BM_DiagDev(N, int, DT_INT32, gpu); \
|
| /external/tensorflow/tensorflow/core/ops/ |
| array_grad.cc | 46 {{"dshape"}, "ZerosLike", {"shape"}, {{"T", DT_INT32}}},
167 {{"d_dim"}, "ZerosLike", {"dim"}, {{"T", DT_INT32}}},
179 {{"T", "$T"}, {"Index", DT_INT32}}});
238 {{"d_dim"}, "ZerosLike", {"dim"}, {{"T", DT_INT32}}},
304 {{"d_dims"}, "ZerosLike", {"dims"}, {{"T", DT_INT32}}},
329 {{"dp"}, "ZerosLike", {"p"}, {{"T", DT_INT32}}},
348 {{"dp"}, "ZerosLike", {"p"}, {{"T", DT_INT32}}},
376 if (itype != DT_INT32) {
400 if (itype != DT_INT32) {
414 {{"b1"}, "ExpandDims", {"begin", "one"}, {{"T", DT_INT32}}},
[all...] |
| array_grad_test.cc | 42 f::NDef("axis", "Placeholder", {}, {{"dtype", DT_INT32}}),
82 f::NDef("axis", "Placeholder", {}, {{"dtype", DT_INT32}}),
121 {f::NDef("dim", "Placeholder", {}, {{"dtype", DT_INT32}}),
127 {"Tin", DataTypeSlice{DT_INT32, T, T, T}},
128 {"Tout", DataTypeSlice{DT_INT32, T, T}}})});
147 f::NDef("dim", "Placeholder", {}, {{"dtype", DT_INT32}}),
151 {"Tin", DataTypeSlice{T, T, DT_INT32, T}},
152 {"Tout", DataTypeSlice{T, T, DT_INT32}}})});
217 {f::NDef("dim", "Placeholder", {}, {{"dtype", DT_INT32}}),
225 {"Tin", DataTypeSlice{DT_INT32, T, T, T}}
[all...] |
| remote_fused_graph_ops_test.cc | 37 src_list.emplace_back("b", 0, DT_INT32);
38 input_types.emplace_back(DT_INT32);
|
| data_flow_ops_test.cc | 96 .Input("indices", 0, DT_INT32)
108 .Input("indices", 0, DT_INT32)
124 .Input({{"indices", 0, DT_INT32}, {"indices_2", 1, DT_INT32}})
168 .Input({{"indices", 0, DT_INT32}, {"indices_2", 1, DT_INT32}})
187 .Input({"size", 0, DT_INT32})
200 .Attr("component_types", {DT_FLOAT, DT_INT32})
222 .Input("n", 0, DT_INT32)
223 .Attr("component_types", {DT_FLOAT, DT_INT32})
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| const_analysis_test.cc | 32 auto arg0 = ops::_Arg(root.WithOpName("Arg0"), DT_INT32, 0);
33 auto arg1 = ops::_Arg(root.WithOpName("Arg1"), DT_INT32, 1);
34 auto arg2 = ops::_Arg(root.WithOpName("Arg2"), DT_INT32, 2);
35 auto arg3 = ops::_Arg(root.WithOpName("Arg3"), DT_INT32, 3);
60 auto arg0 = ops::_Arg(root.WithOpName("Arg0"), DT_INT32, 0);
61 auto arg1 = ops::_Arg(root.WithOpName("Arg1"), DT_INT32, 1);
62 auto arg2 = ops::_Arg(root.WithOpName("Arg2"), DT_INT32, 2);
|
| functionalize_control_flow_test.cc | 68 auto x = ops::Placeholder(scope.WithOpName("x"), DT_INT32);
69 auto y = ops::Placeholder(scope.WithOpName("y"), DT_INT32);
111 auto y = ops::Placeholder(scope.WithOpName("y"), DT_INT32);
112 auto x = ops::Placeholder(scope.WithOpName("x"), DT_INT32);
116 else_fn, {DT_INT32});
126 auto arg_1 = ops::_Arg(scope.WithOpName("_arg1"), DT_INT32, 1);
127 auto arg_2 = ops::_Arg(scope.WithOpName("_arg2"), DT_INT32, 2);
140 EXPECT_EQ(DataTypeVector{DT_INT32}, result.ret_types);
141 EXPECT_EQ((DataTypeVector{DT_BOOL, DT_INT32, DT_INT32}), result.arg_types)
[all...] |
| tf2xla_test.cc | 45 (*x->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32);
49 (*y->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32);
55 (*sum->mutable_attr())["T"] = TypeAttrValue(DT_INT32);
|
| literal_util_test.cc | 33 LiteralToHostTensor(*int64_values_literal, DT_INT32, &host_tensor)
53 LiteralToHostTensor(*int32_values_literal, DT_INT32, &host_tensor)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bcast_ops.cc | 34 OP_REQUIRES_OK(ctx, ctx->MatchSignature({DT_INT32, DT_INT32}, {DT_INT32}));
58 Tensor output(DT_INT32, TensorShape({len}));
82 ctx, ctx->MatchSignature({DT_INT32, DT_INT32}, {DT_INT32, DT_INT32}));
117 Tensor constant(DT_INT32, TensorShape({len}));
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| range_coder_ops_test.cc | 86 .Input(tensorflow::FakeInput(DT_INT32))
110 .Input(tensorflow::FakeInput(DT_INT32))
111 .Input(tensorflow::FakeInput(DT_INT32))
137 Tensor shape{DT_INT32, {data_shape.dims()}};
176 Tensor histogram_tensor{DT_INT32, cdf_tensor->shape()};
210 Tensor temp{DT_INT32, {1, 1, 1, 1, kMaxValue + 2}};
222 Tensor cdf{DT_INT32, {1, 32, 32, 16, kMaxValue + 2}};
246 Tensor cdf{DT_INT32, {1, 1, kDimensionSize, kMaxMaxValue + 2}};
256 Tensor cdf{DT_INT32, {1, kDimensionSize, 1, kMaxMaxValue + 2}};
275 Tensor cdf{DT_INT32, {2, 1, 1, 7, kMaxMaxValue + 2}}
[all...] |
