| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | reduce_utils_test.cc | 26 const std::vector<int>& shape_in, in TestFunction()
49 const std::vector<int> shape_in = {2, 3, 4, 5}; in TEST() local
57 const std::vector<int> shape_in = {2, 3, 4, 5}; in TEST() local
65 const std::vector<int> shape_in = {}; in TEST() local
73 const std::vector<int> shape_in = {4, 5}; in TEST() local
81 const std::vector<int> shape_in = {4, 3}; in TEST() local
89 const std::vector<int> shape_in = {4, 3, 5}; in TEST() local
97 const std::vector<int> shape_in = {4, 3, 2}; in TEST() local
105 const std::vector<int> shape_in = {4, 3, 2}; in TEST() local
113 const std::vector<int> shape_in = {1, 4, 3, 1}; in TEST() local
[all …]
|
| H A D | reduce_utils.h | 59 int& out_num_axis, const int* shape_in, int* shape_out, in ResolveAxis()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | xent_op.cc | 46 TensorShape shape_in = logits_in.shape(); in Compute() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | GEMMReshapeRHSMatrixFixture.h | 54 …void setup(TensorShape shape_in, unsigned int batch_size, DataType data_type, unsigned int n0, uns… in setup()
|
| H A D | GEMMReshapeLHSMatrixFixture.h | 54 …void setup(TensorShape shape_in, unsigned int batch_size, DataType data_type, unsigned int m0, uns… in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference_test.cc | 890 Shape shape_in = ShapeUtil::MakeShape(F32, {16, 8}); in TEST_F() local
929 Shape shape_in = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops.cc | 476 ShapeHandle shape_in = c->input(1); in __anon38bbb0e80702() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties_test.cc | 787 auto shape_in = properties.GetOutputProperties("ones").at(0).shape(); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 10421 def helper(shape_in, shape_out, align_corners): argument
10441 def helper(shape_in, shape_out, align_corners): argument
|