| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_pooling_gpu.cc | 83 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
213 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorDimApply.h | 26 for (const auto dim_i : c10::irange(ndims)) { in tensor_dim_apply3() local
|
| H A D | Convolution.cpp | 839 auto dim_i = inp_view_as_complex.dim() - 1; in complex_to_real() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/ |
| H A D | tensor_view.h | 109 inline int Dim(int dim_i) const { in Dim()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensorMath.cpp | 1186 void s_addmm_out_sparse_dense_worker(int64_t nnz, int64_t dim_i, int64_t dim_j, int64_t dim_k, Tens… in s_addmm_out_sparse_dense_worker()
1280 int64_t dim_i = sparse_.size(0); in s_addmm_out_sparse_dense_cpu() local
1509 int64_t dim_i = sparse.size(0); in _sspaddmm_out_cpu() local
1976 int64_t dim_i = self_coalesced.size(1); in bmm_out_sparse_cpu() local
|
| H A D | SparseCsrTensorMath.cpp | 531 auto dim_i = sparse.size(0); in addmm_out_sparse_csr_native_cpu() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | tensor_format_test.cc | 64 int dim_h, dim_w, dim_i, dim_o; member
|
| H A D | strided_slice_op.cc | 281 int64_t dim_i = input_shape.dim_size(i); in ValidateStridedSliceOp() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | cuda_codegen.cpp | 191 auto dim_i = intValue(dim); in print_flat_alloc() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorMath.cpp | 123 int64_t dim_i = list[i].dim(); in _nested_tensor_from_tensor_list() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 1368 def test_shape(num_mats, dim_i, dim_j, dim_k, nnz): argument
1424 def test_shape(num_mats, dim_i, dim_j, dim_k, nnz): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 5577 Value dim_i = rewriter.create<arith::IndexCastOp>( in matchAndRewrite() local
5585 Value dim_i = rewriter.create<arith::ConstantOp>( in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 1914 int64_t dim_i = input_shape[i]; in CalculateSlicedShapeFromDenseIndices() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 2054 int dim_i = input_type.getShape()[i]; in verify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 640 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch() local
|