Home
last modified time | relevance | path

Searched defs:strided (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseBlasImpl.cpp32 bool operands_support_triton_mm_kernel(const Tensor& compressed, const Tensor& strided) { in operands_support_triton_mm_kernel()
56 Tensor& _compressed_row_strided_mm_out(const Tensor& compressed, const Tensor& strided, Tensor& res… in _compressed_row_strided_mm_out()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DTensorFactories.cpp95 IntArrayRef strided, in empty_strided_unknown_quantized()
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_fs_generator.cpp1294 struct brw_reg strided = stride(suboffset(src[0], component * s), in generate_code() local
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/range/include/boost/range/adaptor/
Dstrided.hpp676 strided = range_detail::forwarder< variable
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_fs_generator.cpp2096 struct elk_reg strided = stride(suboffset(src[0], component * s), in generate_code() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.cc460 auto strided = args.get<runtime::StridedMemrefView>(i); in operator ()() local
1533 if (auto strided = args.get<runtime::StridedMemrefView>(i); in operator ()() local
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_convolution.py2814 strided, argument
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py306 def strided(self): member in InputGen