| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | identity_op_test.cc | 40 TF_ASSERT_OK(Init(DT_INT32));
43 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
49 TF_ASSERT_OK(Init(DT_INT32));
52 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3})); local
|
| H A D | guarantee_const_op_test.cc | 41 TF_ASSERT_OK(Init(DT_INT32));
44 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
50 TF_ASSERT_OK(Init(DT_INT32));
53 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3})); local
69 AddResourceInput("", "resource", new Var(DT_INT32));
|
| H A D | identity_n_op_test.cc | 40 TF_ASSERT_OK(Init(DT_INT32, DT_DOUBLE));
45 Tensor expected0(allocator(), DT_INT32, TensorShape({6})); local
54 TF_ASSERT_OK(Init(DT_INT32, DT_INT32));
58 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3})); local
66 TF_ASSERT_OK(Init(DT_STRING, DT_INT32));
73 Tensor expected1(allocator(), DT_INT32, TensorShape({8})); local
|
| H A D | logging_ops_test.cc | 44 TF_ASSERT_OK(Init(DT_INT32, DT_INT32));
49 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
55 TF_ASSERT_OK(Init(DT_INT32, DT_INT32, "", -1, 6));
60 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
66 TF_ASSERT_OK(Init(DT_INT32, DT_STRING));
71 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
77 TF_ASSERT_OK(Init(DT_INT32, DT_STRING, "Message: "));
82 Tensor expected(allocator(), DT_INT32, TensorShap local
94 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
[all...] |
| H A D | non_max_suppression_op_test.cc | 37 .Input(FakeInput(DT_INT32))
54 Tensor expected(allocator(), DT_INT32, TensorShape({3})); local
68 Tensor expected(allocator(), DT_INT32, TensorShape({3})); local
83 Tensor expected(allocator(), DT_INT32, TensorShape({2})); local
98 Tensor expected(allocator(), DT_INT32, TensorShape({3})); local
110 Tensor expected(allocator(), DT_INT32, TensorShape({1})); local
133 Tensor expected(allocator(), DT_INT32, TensorShape({1})); local
174 Tensor expected(allocator(), DT_INT32, TensorShape({0})); local
189 .Input(FakeInput(DT_INT32))
207 Tensor expected(allocator(), DT_INT32, TensorShap local
223 Tensor expected(allocator(), DT_INT32, TensorShape({3})); local
239 Tensor expected(allocator(), DT_INT32, TensorShape({2})); local
256 Tensor expected(allocator(), DT_INT32, TensorShape({3})); local
269 Tensor expected(allocator(), DT_INT32, TensorShape({1})); local
293 Tensor expected(allocator(), DT_INT32, TensorShape({1})); local
337 Tensor expected(allocator(), DT_INT32, TensorShape({0})); local
[all...] |
| H A D | control_flow_ops_test.cc | 40 Initialize(DT_INT32);
44 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
51 Initialize(DT_INT32);
55 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
62 Initialize(DT_INT32);
66 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3})); local
|
| H A D | cross_op_test.cc | 82 .Input(FakeInput(DT_INT32))
83 .Input(FakeInput(DT_INT32))
94 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3})); local
|
| H A D | sequence_ops_test.cc | 56 MakeOp(DT_INT32);
65 Tensor expected(allocator(), DT_INT32, TensorShape({5})); local
103 MakeOp(DT_FLOAT, DT_INT32);
133 MakeOp(DT_DOUBLE, DT_INT32);
|
| H A D | debug_ops_test.cc | 62 TF_ASSERT_OK(Init(DT_INT32));
65 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
87 TF_ASSERT_OK(Init(DT_INT32, debug_urls));
90 Tensor expected(allocator(), DT_INT32, TensorShape({6})); local
133 Tensor tensor_prime(DT_INT32);
160 TF_ASSERT_OK(Init(DT_INT32));
163 Tensor expected(allocator(), DT_INT32, TensorShape({2, 3})); local
497 TF_ASSERT_OK(Init(DT_INT32));
516 static_cast<double>(DT_INT32), // dtype.
698 TF_ASSERT_OK(Init(DT_INT32));
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| H A D | while_gradients_test.cc | 34 void Init(int num_inputs, DataType dtype = DT_INT32) { argument
|
| /external/tensorflow/tensorflow/cc/ops/ |
| H A D | while_loop_test.cc | 32 void Init(int num_inputs, DataType dtype = DT_INT32) { argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor_shape_test.cc | 178 TensorShapeTestHelper::set_data_type(&s, DT_INT32);
180 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INT32); local
182 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INT32); local
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| H A D | import_tensorflow.cc | 54 using tensorflow::DT_INT32;
137 else if (dtype == DT_INT32)
224 CHECK_EQ(input_tensor.dtype(), DT_INT32);
334 case DT_INT32:
1286 CHECK(indices_data_type == DT_INT32 || indices_data_type == DT_INT64);
1301 CHECK(axis_data_type == DT_INT64 || axis_data_type == DT_INT32);
1302 CHECK(output_type == DT_INT64 || output_type == DT_INT32);
1439 CHECK_EQ(GetDataTypeAttr(node, "Tblock_shape"), DT_INT32); local
1440 CHECK_EQ(GetDataTypeAttr(node, "Tpaddings"), DT_INT32); local
1454 CHECK_EQ(GetDataTypeAttr(node, "Tblock_shape"), DT_INT32); local
1455 CHECK_EQ(GetDataTypeAttr(node, "Tcrops"), DT_INT32); local
[all...] |