Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dmul_op.cc59 auto a_dense_shape = a_dense_shape_t.vec<int64_t>(); in Compute() local
H A Dmat_mul_op.cc132 const auto& a_dense_shape = sparse_matrix_a.dense_shape().vec<int64_t>(); in ValidateInputs() local
520 auto a_dense_shape = a_dense_shape_t.vec<int64_t>(); in Compute() local
H A Dadd_op.cc85 auto a_dense_shape = a.dense_shape().vec<int64_t>(); in operator ()() local
H A Dsparse_mat_mul_op.cc332 auto a_dense_shape = a_matrix->dense_shape().vec<int64_t>(); in Compute() local