Home
last modified time | relevance | path

Searched defs:is_row_major (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
DCUDASparseDescriptors.cpp70 bool is_row_major = at::native::is_blas_compatible_row_major_order(input); in createRawDnMatDescriptor() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc270 bool is_row_major = xla::LayoutUtil::Minor(shape.layout(), row_dim) != 0; in GemmOpConversionRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1158 const auto is_row_major = [](mlir::Value value) { in RowVectorizationEnabled() local
/external/pytorch/torch/
D_meta_registrations.py5603 def is_row_major(stride): function