Searched defs:dense_shape_t (Results 1 – 9 of 9) sorted by relevance
66 Tensor dense_shape_t = logits_matrix->dense_shape(); in Compute() local149 Tensor dense_shape_t = softmax_matrix->dense_shape(); in Compute() local
66 const Tensor& dense_shape_t = csr_sparse_matrix->dense_shape(); in Compute() local139 const Tensor& dense_shape_t = csr_sparse_matrix->dense_shape(); in Compute() local
48 const Tensor& dense_shape_t = c->input(0); in Compute() local
148 const Tensor& dense_shape_t = csr_sparse_matrix->dense_shape(); in Compute() local
143 const Tensor& dense_shape_t = c->input(2); in ComputeAsync() local
249 Tensor dense_shape_t(cpu_allocator(), DT_INT64, TensorShape({rank})); in ComputeAsync() local
457 auto dense_shape_t = dense_shape.vec<int64_t>(); in ValidateTypesAndShapes() local
37 Tensor dense_shape_t = test::AsTensor<int64_t>({5, 6}); in TEST() local
219 const Tensor& dense_shape_t = context->input(kDenseShapeInput); in SparseFillEmptyRowsOpImpl() local