Home
last modified time | relevance | path

Searched defs:row_dim (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dadd_op.cc299 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in GetWorkspaceSize() local
327 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in GetOutputStructure() local
355 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in Compute() local
H A Dsparse_mat_mul_op.cc138 const int row_dim = (rank == 2) ? 0 : 1; in Compute() local
343 const int row_dim = (rank == 2) ? 0 : 1; in Compute() local
H A Dsparse_cholesky_op.cc245 const int row_dim = (rank == 2) ? 0 : 1; in ValidateInputs() local
H A Dmat_mul_op.cc524 const int row_dim = (rank == 2) ? 0 : 1; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Ddense_to_sparse_batch_dataset_op.cc257 for (int64_t row_dim = strides.size() - 2; row_dim >= 0; in GetNextInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
H A Dgemm_pattern.cc226 int64_t row_dim = dim_nums.lhs_batching_dimensions().size(); in GemmOpConversionRewrite() local