Home
last modified time | relevance | path

Searched refs:OP_REQUIRES_OK_ASYNC (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dqueue_op.cc56 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 …]
Dnccl_ops.cc100 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()
Dbarrier_ops.cc131 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 …]
Ddynamic_partition_op_gpu.cu.cc210 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 …]
Dcollective_ops.cc178 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 …]
Dbatch_kernels.cc258 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()
Dconditional_accumulator_base_op.h199 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()
Dstack.cc224 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()
Dfunction_ops.cc220 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()
Dwhere_op.cc281 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()
Dlookup_table_init_op.cc177 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
181 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
Dmutex_ops.cc143 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/
Dmatrix_inverse_op.cc122 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()
Dmatrix_solve_op.cc158 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 …]
Dqr_op_impl.h160 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 …]
Dlu_op_gpu.cu.cc125 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()
Dsvd_op_gpu.cu.cc151 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 …]
Dself_adjoint_eig_v2_op_gpu.cc68 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()
Ddeterminant_op.cc155 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 …]
Dcholesky_op.cc126 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/
Dsparse_tensor_to_csr_sparse_matrix_op.cc146 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()
Ddense_to_csr_sparse_matrix_op.cc190 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/
Dgpu_swapping_kernels.cc41 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/
Dcrop_and_resize_op.cc151 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/
Dbatch_resource_base.cc648 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()

12