| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_striding_sycl.cpp | 37 Eigen::array<IndexType, 4> tensor_dims = {{2,3,5,7}}; in test_simple_striding() local
119 Eigen::array<IndexType, 4> tensor_dims = {{2,3,5,7}}; in test_striding_as_lvalue() local
|
| H A D | cxx11_tensor_block_io.cpp | 222 DSizes<Index, 3> tensor_dims(7, 9, 7); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
269 DSizes<Index, 4> tensor_dims(7, 5, 9, 9); in test_block_io_copy_using_reordered_dimensions_squeeze() local
|
| H A D | cxx11_tensor_block_access.cpp | 73 DSizes<Index, 2> tensor_dims(100, 100); in test_block_mapper_sanity() local
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_spec/ |
| H A D | _internals.py | 114 def check_tensor(shards_metadata, tensor_dims) -> None: argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_util.cc | 71 auto tensor_dims = tensor.flat<int64_t>(); in ShapeTensorToTensorShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantization_utils.cc | 172 RuntimeShape tensor_dims = in FillPerChannelMinMax() local
388 RuntimeShape tensor_dims = in SymmetricPerChannelQuantizeValues() local
|
| H A D | quantize_model.cc | 633 RuntimeShape tensor_dims = in RequantizeConstant() local
|
| /aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/ |
| H A D | test_sharded_tensor.py | 1951 def verify_size(rank, tensor_dims): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/ |
| H A D | util.cpp | 165 std::vector<std::vector<int64_t>> tensor_dims; in flattenList() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate.cc | 514 uint32_t* tensor_dims = reinterpret_cast<uint32_t*>(tensor->dims->data); in ConvertTensorTypeToNNType() local
1533 const uint32_t* tensor_dims = reinterpret_cast<const uint32_t*>(dims->data); in AddNewInputConstantTensor() local
1849 uint32_t* tensor_dims = in AddTensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_test.cc | 2424 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 3702 std::vector<int64_t> tensor_dims(dims.begin(), dims.end()); in ConvertPack() local
|
| H A D | convert_nodes_test.cc | 2915 std::vector<int> tensor_dims; in TEST_P() member
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_linalg.py | 3627 def run_test(tensor_dims, some): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1539 int tensor_dims[] = {1, input_size, 1}; in Create() local
|