| /aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/ |
| H A D | sparse_tensor_test.cc | 36 GetSimpleIndexTensor(int N, const int NDIM) { in GetSimpleIndexTensor()
62 const int NDIM = 3; in TEST() local
95 const int NDIM = 3; in TEST() local
108 const int NDIM = 3; in TEST() local
121 const int NDIM = 3; in TEST() local
134 const int NDIM = 3; in TEST() local
147 const int NDIM = 3; in TEST() local
159 const int NDIM = 3; in TEST() local
172 const int NDIM = 3; in TEST() local
230 const int NDIM = 3; in TEST() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | strided_slice_op_impl.h | 190 #define PREVENT_INSTANTIATE_DIM1_AND_UP(T, NDIM) \ argument
225 #define PREVENT_INSTANTIATE_DIM0_ONLY(T, NDIM) \ argument
238 #define PREVENT_INSTANTIATE(T, NDIM) PREVENT_INSTANTIATE_DIM0_ONLY(T, NDIM) argument
240 #define PREVENT_INSTANTIATE(T, NDIM) PREVENT_INSTANTIATE_DIM1_AND_UP(T, NDIM) argument
243 #define PREVENT_INSTANTIATE(T, NDIM) argument
|
| H A D | tile_ops.cc | 101 #define DECLARE_CUDA_DIM(T, NDIM) \ argument
105 #define DECLARE_CUDA_DIM(T, NDIM) argument
128 #define DECLARE_DIM(T, NDIM) \ argument
373 #define HANDLE_DIM(DT, NDIM) \ in Compute() argument
|
| H A D | slice_op.cc | 195 #define HANDLE_DIM(NDIM) \ in Compute() argument
241 #define DECLARE_CPU_SPEC(T, NDIM) \ argument
281 #define DECLARE_GPU_SPEC(T, NDIM) \ argument
|
| H A D | tile_ops_gpu_impl.h | 41 #define DEFINE_DIM(T, NDIM) \ argument
45 #define DEFINE_TILE_OPS(NDIM) \ argument
|
| H A D | strided_slice_op.cc | 161 #define HANDLE_DIM(NDIM) \ in Compute() argument
260 #define HANDLE_DIM(NDIM) \ in Compute() argument
382 #define HANDLE_DIM(NDIM) \ in Compute() argument
|
| H A D | betainc_op.cc | 91 #define CASE(NDIM) \ in Compute() argument
129 #define DECLARE_GPU_SPEC_NDIM(T, NDIM) \ argument
|
| H A D | where_op.cc | 159 #define HANDLE_DIM(NDIM) \ in Compute() argument
312 #define HANDLE_DIM(NDIM) \ in ComputeAsyncType() argument
|
| H A D | betainc_op_gpu.cu.cc | 33 #define DEFINE_GPU_KERNELS_NDIM(T, NDIM) \ argument
|
| H A D | tile_ops_cpu_impl.h | 30 #define DEFINE_DIM(T, NDIM) \ argument
|
| H A D | sparse_dense_binary_op_shared.cc | 145 #define CASE(NDIM) \ in Compute() argument
|
| H A D | gather_nd_op.cc | 81 #define DECLARE_GPU_SPECS_INDEX_NDIM(T, Index, NDIM) \ argument
|
| H A D | argmax_op.cc | 84 #define HANDLE_DIM(NDIM) \ in Compute() argument
|
| H A D | gather_nd_op_gpu.cu.cc | 104 #define DEFINE_GPU_SPECS_INDEX_NDIM(T, Index, NDIM) \ argument
|
| H A D | sparse_to_dense_op_test.cc | 202 const int NDIM = state.range(0); in BM_SparseToDense() local
|
| H A D | reverse_sequence_op.cc | 144 #define HANDLE_DIM(NDIM) \ in Compute() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | einsum_op_gpu.cu.cc | 24 #define DECLARE_GPU_SPECS_NDIM(T, NDIM) \ argument
|