Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc522 const int64 b_slice_size = b_inner_dim * b_outer_dim; in Compute() local
579 const T* b_i = b_base_ptr + i * b_slice_size; in Compute()
667 typename TTypes<T>::UnalignedConstMatrix b_i(b.data() + i * b_slice_size, in Compute()