| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op.cc | 99 OP_REQUIRES_ASYNC( in RunIfBoxIndexIsValid()
139 OP_REQUIRES_ASYNC(context, image.dims() == 4, in ComputeAsync()
147 OP_REQUIRES_ASYNC( in ComputeAsync()
154 OP_REQUIRES_ASYNC(context, crop_size.dims() == 1, in ComputeAsync()
158 OP_REQUIRES_ASYNC( in ComputeAsync()
168 OP_REQUIRES_ASYNC( in ComputeAsync()
373 OP_REQUIRES_ASYNC(context, grads.dims() == 4, in ComputeAsync()
379 OP_REQUIRES_ASYNC( in ComputeAsync()
385 OP_REQUIRES_ASYNC( in ComputeAsync()
390 OP_REQUIRES_ASYNC(context, image_size.dims() == 1, in ComputeAsync()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 62 OP_REQUIRES_ASYNC( in ComputeAsync()
77 OP_REQUIRES_ASYNC(c, !already_cancelled, in ComputeAsync()
364 OP_REQUIRES_ASYNC( in ComputeAsyncImpl()
502 OP_REQUIRES_ASYNC( in ComputeAsync()
512 OP_REQUIRES_ASYNC( in ComputeAsync()
515 OP_REQUIRES_ASYNC( in ComputeAsync()
518 OP_REQUIRES_ASYNC( in ComputeAsync()
631 OP_REQUIRES_ASYNC( in ComputeAsync()
641 OP_REQUIRES_ASYNC(c, group_size.dims() == 0, in ComputeAsync()
646 OP_REQUIRES_ASYNC(c, group_key.dims() == 0, in ComputeAsync()
[all …]
|
| D | barrier_ops.cc | 93 OP_REQUIRES_ASYNC( in TryInsertMany()
113 OP_REQUIRES_ASYNC( in TryInsertMany()
143 OP_REQUIRES_ASYNC( in TryInsertMany()
534 OP_REQUIRES_ASYNC( in ComputeAsync()
586 OP_REQUIRES_ASYNC(ctx, TensorShapeUtils::IsScalar(Tnum_elements->shape()), in ComputeAsync()
|
| D | debug_ops.h | 718 OP_REQUIRES_ASYNC(
734 OP_REQUIRES_ASYNC(context, stream != nullptr,
761 OP_REQUIRES_ASYNC(context, stream != nullptr,
788 OP_REQUIRES_ASYNC(context, stream != nullptr,
820 OP_REQUIRES_ASYNC(context, stream != nullptr,
853 OP_REQUIRES_ASYNC(context, stream != nullptr,
|
| D | queue_op.cc | 214 OP_REQUIRES_ASYNC(ctx, num_elements >= 0, in ComputeAsync()
287 OP_REQUIRES_ASYNC(ctx, num_elements >= 0, in ComputeAsync()
|
| D | dynamic_partition_op_gpu.cu.cc | 239 OP_REQUIRES_ASYNC( in ComputeAsync()
285 OP_REQUIRES_ASYNC(c, stream, errors::Internal("No GPU stream available."), in ComputeAsync()
303 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | check_numerics_op.cc | 223 OP_REQUIRES_ASYNC(context, stream != nullptr, in ComputeAsync()
247 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | where_op.cc | 299 OP_REQUIRES_ASYNC( in ComputeAsyncType()
347 OP_REQUIRES_ASYNC( in ComputeAsyncType()
|
| D | function_ops.cc | 215 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
273 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
|
| D | segment_reduction_ops_impl.h | 235 OP_REQUIRES_ASYNC( in ComputeAsync()
240 OP_REQUIRES_ASYNC( in ComputeAsync()
264 OP_REQUIRES_ASYNC( in ComputeAsync()
284 OP_REQUIRES_ASYNC(context, output_rows > 0, in ComputeAsync()
|
| D | partitioned_function_ops.cc | 88 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
|
| D | sendrecv_ops.cc | 191 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_solve_op.cc | 135 OP_REQUIRES_ASYNC( in ComputeAsync()
139 OP_REQUIRES_ASYNC(context, rhs.dims() == ndims, in ComputeAsync()
144 OP_REQUIRES_ASYNC( in ComputeAsync()
149 OP_REQUIRES_ASYNC(context, rhs.dim_size(ndims - 2) == n, in ComputeAsync()
303 OP_REQUIRES_ASYNC( in ComputeAsync()
345 OP_REQUIRES_ASYNC(context, host_infos[0].data()[i] <= 0, in ComputeAsync()
|
| D | determinant_op.cc | 138 OP_REQUIRES_ASYNC( in ComputeAsync()
142 OP_REQUIRES_ASYNC( in ComputeAsync()
252 OP_REQUIRES_ASYNC( in ComputeAsync()
278 OP_REQUIRES_ASYNC( in ComputeAsync()
282 OP_REQUIRES_ASYNC( in ComputeAsync()
399 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | cholesky_op.cc | 104 OP_REQUIRES_ASYNC( in ComputeAsync()
108 OP_REQUIRES_ASYNC( in ComputeAsync()
201 OP_REQUIRES_ASYNC(context, status.ok(), errors::InvalidArgument(kErrMsg), in ComputeAsync()
|
| D | lu_op_gpu.cu.cc | 89 OP_REQUIRES_ASYNC( in ComputeAsync()
97 OP_REQUIRES_ASYNC( in ComputeAsync()
245 OP_REQUIRES_ASYNC(context, host_infos[0].data()[i] <= 0, in ComputeAsync()
|
| D | matrix_inverse_op.cc | 103 OP_REQUIRES_ASYNC( in ComputeAsync()
107 OP_REQUIRES_ASYNC( in ComputeAsync()
252 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | self_adjoint_eig_v2_op_gpu.cc | 51 OP_REQUIRES_ASYNC( in ComputeAsync()
56 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | qr_op_impl.h | 149 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 79 OP_REQUIRES_ASYNC( in ComputeAsync()
85 OP_REQUIRES_ASYNC( in ComputeAsync()
90 OP_REQUIRES_ASYNC( in ComputeAsync()
102 OP_REQUIRES_ASYNC( in ComputeAsync()
110 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | transfer_ops.cc | 57 OP_REQUIRES_ASYNC(ctx, !already_cancelled, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | dense_to_csr_sparse_matrix_op.cc | 163 OP_REQUIRES_ASYNC(c, rank == 2 || rank == 3, in ComputeAsync()
168 OP_REQUIRES_ASYNC( in ComputeAsync()
173 OP_REQUIRES_ASYNC( in ComputeAsync()
206 OP_REQUIRES_ASYNC( in ComputeAsync()
243 OP_REQUIRES_ASYNC( in ComputeAsync()
266 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 129 OP_REQUIRES_ASYNC( in ComputeAsync()
162 OP_REQUIRES_ASYNC( in ComputeAsync()
199 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 31 OP_REQUIRES_ASYNC( in ComputeAsync()
64 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_requires.h | 72 #define OP_REQUIRES_ASYNC(CTX, EXP, STATUS, CALLBACK) \ macro
|