Searched refs:output_csr_matrix_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | sparse_cholesky_op.cc | 220 Tensor* output_csr_matrix_tensor; in Compute() local
224 ctx, ctx->allocate_output(0, TensorShape({}), &output_csr_matrix_tensor, in Compute()
226 output_csr_matrix_tensor->scalar<Variant>()() = in Compute()
|
| H A D | sparse_tensor_to_csr_sparse_matrix_op.cc | 118 Tensor* output_csr_matrix_tensor; in Compute() local
122 ctx, ctx->allocate_output(0, TensorShape({}), &output_csr_matrix_tensor, in Compute()
124 output_csr_matrix_tensor->scalar<Variant>()() = in Compute()
|
| H A D | dense_to_csr_sparse_matrix_op.cc | 123 Tensor* output_csr_matrix_tensor; in Compute() local
127 ctx, ctx->allocate_output(0, TensorShape({}), &output_csr_matrix_tensor, in Compute()
129 output_csr_matrix_tensor->scalar<Variant>()() = in Compute()
|
| H A D | sparse_mat_mul_op.cc | 252 Tensor* output_csr_matrix_tensor; in Compute() local
256 ctx, ctx->allocate_output(0, TensorShape({}), &output_csr_matrix_tensor, in Compute()
258 output_csr_matrix_tensor->scalar<Variant>()() = in Compute()
|