Searched refs:ExtractVariantFromInput (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | softmax_op.cc | 50 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &logits_matrix)); in Compute()
122 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &softmax_matrix)); in Compute()
130 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 1, &grad_softmax_matrix)); in Compute()
|
| D | nnz_op.cc | 48 OP_REQUIRES_OK(c, ExtractVariantFromInput(c, 0, &csr_sparse_matrix)); in Compute()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 74 OP_REQUIRES_OK(c, ExtractVariantFromInput(c, 0, &csr_sparse_matrix)); in Compute()
143 OP_REQUIRES_OK(c, ExtractVariantFromInput(c, 0, &csr_sparse_matrix)); in Compute()
|
| D | csr_sparse_matrix_to_dense_op.cc | 57 ExtractVariantFromInput(context, 0, &csr_sparse_matrix)); in Compute()
129 OP_REQUIRES_OK(c, ExtractVariantFromInput(c, 0, &csr_sparse_matrix)); in Compute()
|
| D | sparse_ordering_amd_op.cc | 69 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &input_matrix)); in Compute()
|
| D | sparse_mat_mul_op.cc | 108 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &input_matrix_a)); in Compute()
109 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 1, &input_matrix_b)); in Compute()
317 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
318 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 1, &b_matrix)); in Compute()
|
| D | mul_op.cc | 47 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
|
| D | sparse_matrix_components_op.cc | 49 OP_REQUIRES_OK(c, ExtractVariantFromInput(c, 0, &csr_sparse_matrix)); in Compute()
|
| D | add_op.cc | 204 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
205 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 1, &b_matrix)); in Compute()
|
| D | sparse_cholesky_op.cc | 80 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &input_matrix)); in Compute()
|
| D | transpose_op.cc | 103 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &input_matrix)); in Compute()
|
| D | mat_mul_op.cc | 171 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &sparse_matrix_a)); in Compute()
501 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
|
| D | sparse_matrix.h | 633 Status ExtractVariantFromInput(OpKernelContext* ctx, int index, in ExtractVariantFromInput() function
|