/external/tensorflow/tensorflow/core/kernels/ |
D | queue_op.cc | 56 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 59 OP_REQUIRES_OK_ASYNC(ctx, GetResourceFromContext(ctx, "handle", &queue), in ComputeAsync() 99 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}), callback); in ComputeAsync() 103 OP_REQUIRES_OK_ASYNC(ctx, ctx->input_list("components", &components), in ComputeAsync() 109 OP_REQUIRES_OK_ASYNC(ctx, queue->ValidateTuple(tuple), callback); in ComputeAsync() 140 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}), callback); in ComputeAsync() 144 OP_REQUIRES_OK_ASYNC(ctx, ctx->input_list("components", &components), in ComputeAsync() 150 OP_REQUIRES_OK_ASYNC(ctx, queue->ValidateManyTuple(tuple), callback); in ComputeAsync() 169 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 173 OP_REQUIRES_OK_ASYNC( in ComputeAsync() [all …]
|
D | nccl_ops.cc | 100 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 104 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync() 135 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync() 167 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, input->shape(), &output), in ComputeAsync() 171 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync() 205 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync() 235 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 238 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, shape, &output), done); in ComputeAsync() 241 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
|
D | barrier_ops.cc | 131 OP_REQUIRES_OK_ASYNC( in TryInsertMany() 166 OP_REQUIRES_OK_ASYNC( in TryInsertMany() 509 OP_REQUIRES_OK_ASYNC(ctx, GetResourceFromContext(ctx, "handle", &barrier), in ComputeAsync() 540 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 549 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("keys", &keys), callback); in ComputeAsync() 550 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("values", &values), callback); in ComputeAsync() 584 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("num_elements", &Tnum_elements), in ComputeAsync() 597 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 610 OP_REQUIRES_OK_ASYNC(ctx, ctx->set_output("indices", indices), in ComputeAsync() 612 OP_REQUIRES_OK_ASYNC(ctx, ctx->set_output("keys", keys), callback); in ComputeAsync() [all …]
|
D | dynamic_partition_op_gpu.cu.cc | 210 OP_REQUIRES_OK_ASYNC( in AllocateTempSpace() 212 OP_REQUIRES_OK_ASYNC( in AllocateTempSpace() 214 OP_REQUIRES_OK_ASYNC( in AllocateTempSpace() 223 OP_REQUIRES_OK_ASYNC(c, c->output_list("outputs", Tout), done); in AllocateOutputs() 231 OP_REQUIRES_OK_ASYNC(c, Tout->allocate(p, shape, &out), done); in AllocateOutputs() 254 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 269 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 291 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 355 OP_REQUIRES_OK_ASYNC( in RadixSort() 392 OP_REQUIRES_OK_ASYNC( in CountAndSortParts() [all …]
|
D | collective_ops.cc | 178 OP_REQUIRES_OK_ASYNC( in ComputeAsyncImpl() 189 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsyncImpl() 283 OP_REQUIRES_OK_ASYNC(c, in ComputeAsyncImpl() 297 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsyncImpl() 358 OP_REQUIRES_OK_ASYNC(c, in ComputeAsyncImpl() 376 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsyncImpl() 435 OP_REQUIRES_OK_ASYNC( in ComputeAsyncImpl() 446 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsyncImpl() 548 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 576 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync() [all …]
|
D | batch_kernels.cc | 258 OP_REQUIRES_OK_ASYNC(c, GetOrCreateFunctionHandle(c, &handle), done); in ComputeAsync() 309 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync() 316 OP_REQUIRES_OK_ASYNC(c, status, done); in ComputeAsync() 491 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync() 498 OP_REQUIRES_OK_ASYNC(c, status, done); in ComputeAsync() 742 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync() 748 OP_REQUIRES_OK_ASYNC(c, status, done); in ComputeAsync() 928 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync() 934 OP_REQUIRES_OK_ASYNC(c, status, done); in ComputeAsync()
|
D | conditional_accumulator_base_op.h | 199 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 219 OP_REQUIRES_OK_ASYNC(ctx, in CheckSignature() 246 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("num_required", &num_required_tensor), in ComputeAsync()
|
D | stack.cc | 224 OP_REQUIRES_OK_ASYNC(ctx, GetStack(ctx, &stack), done); in ComputeAsync() 278 OP_REQUIRES_OK_ASYNC(ctx, stack->Push({tensor, alloc_attrs, false}), done); in ComputeAsync() 293 OP_REQUIRES_OK_ASYNC(ctx, GetStack(ctx, &stack), done); in ComputeAsync() 299 OP_REQUIRES_OK_ASYNC(ctx, stack->Pop(&value), done); in ComputeAsync()
|
D | function_ops.cc | 220 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 278 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("target", &target), done); in ComputeAsync() 281 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 313 OP_REQUIRES_OK_ASYNC(ctx, in ComputeAsync() 325 OP_REQUIRES_OK_ASYNC(ctx, ctx->input_list("args", &arguments), done); in ComputeAsync()
|
D | where_op.cc | 281 OP_REQUIRES_OK_ASYNC(context, in ComputeAsyncType() 293 OP_REQUIRES_OK_ASYNC(context, s, done); in ComputeAsyncType() 323 OP_REQUIRES_OK_ASYNC(context, in ComputeAsyncType() 333 OP_REQUIRES_OK_ASYNC(context, s, done); \ in ComputeAsyncType()
|
D | lookup_table_init_op.cc | 177 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 181 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
|
D | mutex_ops.cc | 143 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 154 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, TensorShape({}), &variant), in ComputeAsync()
|
/external/tensorflow/tensorflow/core/kernels/linalg/ |
D | matrix_inverse_op.cc | 122 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 133 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 145 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 152 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 185 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 197 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 205 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 219 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 233 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 258 OP_REQUIRES_OK_ASYNC(context, status, done); in ComputeAsync()
|
D | matrix_solve_op.cc | 158 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 180 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 185 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 188 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 204 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 230 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 240 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 256 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 262 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 296 OP_REQUIRES_OK_ASYNC( in ComputeAsync() [all …]
|
D | qr_op_impl.h | 160 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(0, q_shape, &q), in ComputeAsync() 165 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(1, r_shape, &r), in ComputeAsync() 182 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 189 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 198 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 209 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 221 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 228 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 258 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 276 OP_REQUIRES_OK_ASYNC( in ComputeAsync() [all …]
|
D | lu_op_gpu.cu.cc | 125 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 137 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 150 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 163 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 172 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 194 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 205 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 216 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 250 OP_REQUIRES_OK_ASYNC(context, status, done); in ComputeAsync()
|
D | svd_op_gpu.cu.cc | 151 OP_REQUIRES_OK_ASYNC( in RunSVD() 155 OP_REQUIRES_OK_ASYNC( in RunSVD() 176 OP_REQUIRES_OK_ASYNC(context, in RunSVD() 189 OP_REQUIRES_OK_ASYNC( in RunSVD() 218 OP_REQUIRES_OK_ASYNC( in RunSVD() 258 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, u_copy, U), done); in RunSVD() 260 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, v_copy, V), in RunSVD() 278 OP_REQUIRES_OK_ASYNC(context, full_status, done); in CheckResult() 301 OP_REQUIRES_OK_ASYNC( in PerformSVD_MgeqN() 306 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, M, &input_copy), in PerformSVD_MgeqN() [all …]
|
D | self_adjoint_eig_v2_op_gpu.cc | 68 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 74 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 91 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 99 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 131 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 151 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
|
D | determinant_op.cc | 155 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(0, out_shape, &out), in ComputeAsync() 173 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 187 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 214 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 224 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 295 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(0, out_shape, &sign), in ComputeAsync() 298 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 318 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 332 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 358 OP_REQUIRES_OK_ASYNC( in ComputeAsync() [all …]
|
D | cholesky_op.cc | 126 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 166 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync() 186 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync()
|
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | sparse_tensor_to_csr_sparse_matrix_op.cc | 146 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync() 155 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 212 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 216 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 220 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 257 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 263 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 272 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
|
D | dense_to_csr_sparse_matrix_op.cc | 190 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync() 199 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 237 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync() 279 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 283 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 287 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 324 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 330 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 339 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | gpu_swapping_kernels.cc | 41 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 71 OP_REQUIRES_OK_ASYNC(ctx, ctx->allocate_output(0, input.shape(), &output), in ComputeAsync()
|
/external/tensorflow/tensorflow/core/kernels/image/ |
D | crop_and_resize_op.cc | 151 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 174 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 383 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 412 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 605 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 615 OP_REQUIRES_OK_ASYNC( in ComputeAsync() 809 OP_REQUIRES_OK_ASYNC( in RunIfBoxIndexIsValid() 831 OP_REQUIRES_OK_ASYNC( in RunIfBoxIndexIsValid()
|
/external/tensorflow/tensorflow/core/kernels/batching_util/ |
D | batch_resource_base.cc | 648 OP_REQUIRES_OK_ASYNC(last_task_context, ValidateBatch(*batch), in ProcessBatch() 656 OP_REQUIRES_OK_ASYNC(last_task_context, concat_status, last_task_callback); in ProcessBatch() 668 OP_REQUIRES_OK_ASYNC( in ProcessBatch() 678 OP_REQUIRES_OK_ASYNC( in ProcessBatch() 687 OP_REQUIRES_OK_ASYNC(task.context, in ProcessBatch() 693 OP_REQUIRES_OK_ASYNC( in ProcessBatch()
|