Searched refs:Int32Tensor (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_function_test.cc | 432 Run({{func_feed, Int32Tensor(3)}}, func_op, -3); in TEST_F()
453 Run({{func_feed, Int32Tensor(3)}}, func_op, -3); in TEST_F()
474 Run({{func_feed, Int32Tensor(3)}}, func_op, -3); in TEST_F()
493 Run({{func_feed, Int32Tensor(3)}}, func_op, 3); in TEST_F()
518 Run({{func_feed, Int32Tensor(3)}}, {{func_op, 0}, {func_op, 1}}, {3, 2}); in TEST_F()
543 Run({{func_feed, Int32Tensor(3)}}, {{func_op, 0}, {func_op, 1}}, {3, 2}); in TEST_F()
566 Run({{func_feed, Int32Tensor(3)}}, func_op, 2 + 3); in TEST_F()
618 Run({{func_feed, Int32Tensor(3)}}, func_op, 2 + 10 + 3); in TEST_F()
649 Run({{func_feed, Int32Tensor(3)}}, {{func_op, 0}, {func_op, 1}}, {5, 5}); in TEST_F()
678 Run({{func_feed, Int32Tensor(3)}}, {{func_op, 0}, {func_op, 1}}, {5, 5}); in TEST_F()
[all …]
|
| H A D | c_test_util.h | 40 TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
44 TF_Tensor* Int32Tensor(const std::vector<int32_t>& values);
46 TF_Tensor* Int32Tensor(int32_t v);
|
| H A D | c_test_util.cc | 64 TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims, in Int32Tensor() function
76 TF_Tensor* Int32Tensor(const std::vector<int32_t>& values) { in Int32Tensor() function
78 return Int32Tensor(&dims, 1, values.data()); in Int32Tensor()
81 TF_Tensor* Int32Tensor(int32_t v) { in Int32Tensor() function
154 unique_tensor_ptr tensor(Int32Tensor(v), TF_DeleteTensor); in ScalarConst()
|
| H A D | c_api_experimental_test.cc | 200 TF_Tensor* tensor_1X6 = Int32Tensor({1, 6}); in TEST_F()
202 TF_Tensor* tensor_1X1X6 = Int32Tensor({1, 1, 6}); in TEST_F()
|
| H A D | c_api_test.cc | 1001 csession.SetInputs({{feed, Int32Tensor(3)}}); in TEST()
1018 csession.SetInputs({{feed, Int32Tensor(7)}}); in TEST()
1062 csession.SetInputs({{feed, Int32Tensor({3, 2, 5})}}); in RunMinTest()
1140 TF_Tensor* feedValues1[] = {Int32Tensor(1)}; in TEST()
1152 TF_Tensor* feedValues2[] = {Int32Tensor(4)}; in TEST()
|
| H A D | while_loop_test.cc | 84 inputs[i] = {inputs_[i].oper, Int32Tensor(v)}; in Run()
|