Searched defs:host_dense_shape (Results 1 – 2 of 2) sorted by relevance
148 const GPUDevice& d, TTypes<int64>::ConstVec host_dense_shape, in operator ()()218 OpKernelContext* c, TTypes<int64>::ConstVec host_dense_shape, in operator ()()300 auto host_dense_shape = a.dense_shape().vec<int64>(); in CSRSparseMatrixBatchMulVecImpl() local422 auto host_dense_shape = logits.dense_shape().vec<int64>(); in CSRSparseMatrixSoftmaxGPUImpl() local611 auto host_dense_shape = softmax.dense_shape().vec<int64>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local
150 auto host_dense_shape = in Compute() local