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() local138 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
147 const Tensor& dense_shape_t = csr_sparse_matrix->dense_shape(); in Compute() local
126 const Tensor& dense_shape_t = c->input(2); in ComputeAsync() local
248 Tensor dense_shape_t(cpu_allocator(), DT_INT64, TensorShape({rank})); in ComputeAsync() local
457 auto dense_shape_t = dense_shape.vec<int64>(); in ValidateTypesAndShapes() local
55 const Tensor& dense_shape_t = context->input(kDenseShapeInput); in Compute() local
37 Tensor dense_shape_t = test::AsTensor<int64>({5, 6}); in TEST() local