Searched defs:sizeC (Results 1 – 5 of 5) sorted by relevance
| /external/eigen/bench/tensors/ |
| D | tensor_contract_sycl_bench.cc | 69 Eigen::array<TensorIndex, 2> sizeC; in contraction() local
123 Eigen::array<TensorIndex, 2> sizeC; in contractionRowMajor() local
174 Eigen::array<TensorIndex, 2> sizeC; in contractionAT() local
226 Eigen::array<TensorIndex, 2> sizeC; in contractionBT() local
278 Eigen::array<TensorIndex, 2> sizeC; in contractionABT() local
|
| D | tensor_benchmarks.h | 533 Eigen::array<TensorIndex, 2> sizeC; in contraction() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | AdaptiveAveragePooling.cpp | 62 int64_t sizeC, in adaptive_avg_pool_single_out_frame()
164 const int64_t sizeC = input.size(-(DIM+1)); in get_output_shape() local
186 int64_t sizeC = input.size(-(kSpatialDim + 1)); in _adaptive_avg_pool() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | AdaptiveAveragePooling.cu | 469 int sizeC = input_.size(1); in adaptive_avg_pool2d_out_cuda_template() local
621 int sizeC = input.size(1); in adaptive_avg_pool2d_backward_out_cuda_template() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 1817 int64_t sizeC,
1927 int64_t sizeC,
1961 int64_t sizeC,
|