Home
last modified time | relevance | path

Searched refs:OP_REQUIRES_OK_ASYNC (Results 1 – 25 of 41) 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 …]
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 …]
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()
Dlu_op_gpu.cu.cc123 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync()
135 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync()
148 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
161 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync()
170 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
192 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
203 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
214 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
248 OP_REQUIRES_OK_ASYNC(context, status, done); in ComputeAsync()
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 …]
Dnccl_ops.cc96 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, input->shape(), &output), in ComputeAsync()
100 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
132 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
165 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, input->shape(), &output), in ComputeAsync()
169 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
204 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()
Dsvd_op_gpu.cu.cc116 OP_REQUIRES_OK_ASYNC( in RunSVD()
142 OP_REQUIRES_OK_ASYNC(context, in RunSVD()
174 OP_REQUIRES_OK_ASYNC( in RunSVD()
211 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, u_copy, U), done); in RunSVD()
227 OP_REQUIRES_OK_ASYNC(context, full_status, done); in CheckResult()
250 OP_REQUIRES_OK_ASYNC( in PerformSVD_MgeqN()
255 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, M, &input_copy), in PerformSVD_MgeqN()
275 OP_REQUIRES_OK_ASYNC( in PerformSVD_MlessN()
334 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(0, shapeS, &outputS), in ComputeAsync()
336 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(1, shapeU, &outputU), in ComputeAsync()
[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 …]
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.cc203 OP_REQUIRES_OK_ASYNC( in AllocateTempSpace()
205 OP_REQUIRES_OK_ASYNC( in AllocateTempSpace()
207 OP_REQUIRES_OK_ASYNC( in AllocateTempSpace()
216 OP_REQUIRES_OK_ASYNC(c, c->output_list("outputs", Tout), done); in AllocateOutputs()
224 OP_REQUIRES_OK_ASYNC(c, Tout->allocate(p, shape, &out), done); in AllocateOutputs()
247 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
262 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
284 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
345 OP_REQUIRES_OK_ASYNC( in RadixSort()
382 OP_REQUIRES_OK_ASYNC( in CountAndSortParts()
[all …]
Dcollective_ops.cc102 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
107 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
200 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync()
210 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
265 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync()
279 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
334 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
341 OP_REQUIRES_OK_ASYNC(c, s, done); in ComputeAsync()
Drpc_op.cc92 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
100 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
102 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
Dbatch_kernels.cc566 OP_REQUIRES_OK_ASYNC(last_task_context, ValidateBatch(*batch), in ProcessBatch()
574 OP_REQUIRES_OK_ASYNC(last_task_context, concat_status, last_task_callback); in ProcessBatch()
586 OP_REQUIRES_OK_ASYNC( in ProcessBatch()
597 OP_REQUIRES_OK_ASYNC( in ProcessBatch()
606 OP_REQUIRES_OK_ASYNC(task.context, in ProcessBatch()
612 OP_REQUIRES_OK_ASYNC( in ProcessBatch()
732 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync()
739 OP_REQUIRES_OK_ASYNC(c, status, done); in ComputeAsync()
812 OP_REQUIRES_OK_ASYNC(c, in ComputeAsync()
819 OP_REQUIRES_OK_ASYNC(c, status, done); in ComputeAsync()
[all …]
Dstack.cc223 OP_REQUIRES_OK_ASYNC(ctx, GetStack(ctx, &stack), done); in ComputeAsync()
277 OP_REQUIRES_OK_ASYNC(ctx, stack->Push({tensor, alloc_attrs, false}), done); in ComputeAsync()
292 OP_REQUIRES_OK_ASYNC(ctx, GetStack(ctx, &stack), done); in ComputeAsync()
298 OP_REQUIRES_OK_ASYNC(ctx, stack->Pop(&value), done); in ComputeAsync()
Dwhere_op.cc276 OP_REQUIRES_OK_ASYNC(context, in ComputeAsyncType()
288 OP_REQUIRES_OK_ASYNC(context, s, done); in ComputeAsyncType()
318 OP_REQUIRES_OK_ASYNC(context, in ComputeAsyncType()
328 OP_REQUIRES_OK_ASYNC(context, s, done); \ in ComputeAsyncType()
Dfunction_ops.cc248 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
310 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("target", &target), done); in ComputeAsync()
312 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
334 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
347 OP_REQUIRES_OK_ASYNC(ctx, ctx->input_list("args", &arguments), done); in ComputeAsync()
Dcrop_and_resize_op.cc145 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
168 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
365 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
394 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
563 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
573 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
767 OP_REQUIRES_OK_ASYNC( in RunIfBoxIndexIsValid()
789 OP_REQUIRES_OK_ASYNC( in RunIfBoxIndexIsValid()
Dcholesky_op.cc127 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync()
150 OP_REQUIRES_OK_ASYNC(context, in ComputeAsync()
Dfunctional_ops.cc152 OP_REQUIRES_OK_ASYNC(ctx, Instantiate(lib, then_func_, &then_handle), done); in ComputeAsync()
153 OP_REQUIRES_OK_ASYNC(ctx, Instantiate(lib, else_func_, &else_handle), done); in ComputeAsync()
239 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
346 OP_REQUIRES_OK_ASYNC(ctx, Instantiate(lib, cond_func_, &cond_handle), done); in ComputeAsync()
347 OP_REQUIRES_OK_ASYNC(ctx, Instantiate(lib, body_func_, &body_handle), done); in ComputeAsync()
Dconditional_accumulator_base_op.h131 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
223 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("num_required", &num_required_tensor), in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc51 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
54 OP_REQUIRES_OK_ASYNC(ctx, in ComputeAsync()
59 OP_REQUIRES_OK_ASYNC(ctx, ctx->env()->NewWritableFile(filename, &file), in ComputeAsync()
67 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
76 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
85 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
90 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_kernels.cc194 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
201 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("columns", &columns_tensor), done); in ComputeAsync()
222 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
226 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
233 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
241 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
252 OP_REQUIRES_OK_ASYNC(ctx, in ComputeAsync()
257 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc576 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
588 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
656 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
659 OP_REQUIRES_OK_ASYNC(ctx, ctx->input_list("initial_state", &inputs), in ComputeAsync()
664 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
679 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
685 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
912 OP_REQUIRES_OK_ASYNC(ctx, ctx->allocate_output(0, TensorShape({}), &handle), in ProduceOutput()
924 OP_REQUIRES_OK_ASYNC(ctx, s, done); in ProduceOutput()
949 OP_REQUIRES_OK_ASYNC( in ComputeAsync()
[all …]
/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()

12