Home
last modified time | relevance | path

Searched refs:a_input_dense_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_mat_mul_op.cc403 auto a_input_dense_shape = a_input_matrix->dense_shape().vec<int64>(); in Compute() local
426 a_input_dense_shape}; in Compute()
460 a_input_dense_shape}; in Compute()
Dmat_mul_op.cc627 auto a_input_dense_shape = a_input_matrix->dense_shape().vec<int64>(); in Compute() local
654 a_input_dense_shape.data() + row_dim, 2); in Compute()