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