| /external/tensorflow/tensorflow/core/kernels/ |
| D | crop_and_resize_op.cc | 93 OP_REQUIRES_ASYNC( in RunIfBoxIndexIsValid()
133 OP_REQUIRES_ASYNC(context, image.dims() == 4, in ComputeAsync()
141 OP_REQUIRES_ASYNC( in ComputeAsync()
148 OP_REQUIRES_ASYNC(context, crop_size.dims() == 1, in ComputeAsync()
152 OP_REQUIRES_ASYNC( in ComputeAsync()
162 OP_REQUIRES_ASYNC( in ComputeAsync()
355 OP_REQUIRES_ASYNC(context, grads.dims() == 4, in ComputeAsync()
361 OP_REQUIRES_ASYNC( in ComputeAsync()
367 OP_REQUIRES_ASYNC( in ComputeAsync()
372 OP_REQUIRES_ASYNC(context, image_size.dims() == 1, in ComputeAsync()
[all …]
|
| D | rpc_op.cc | 66 OP_REQUIRES_ASYNC( in ComputeAsync()
69 OP_REQUIRES_ASYNC( in ComputeAsync()
72 OP_REQUIRES_ASYNC( in ComputeAsync()
79 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| 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 | cholesky_op.cc | 105 OP_REQUIRES_ASYNC( in ComputeAsync()
109 OP_REQUIRES_ASYNC( in ComputeAsync()
161 OP_REQUIRES_ASYNC(context, status.ok(), errors::InvalidArgument(kErrMsg), 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 | collective_ops.cc | 89 OP_REQUIRES_ASYNC( in ComputeAsync()
187 OP_REQUIRES_ASYNC( in ComputeAsync()
252 OP_REQUIRES_ASYNC( in ComputeAsync()
271 OP_REQUIRES_ASYNC( in ComputeAsync()
321 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | lu_op_gpu.cu.cc | 87 OP_REQUIRES_ASYNC( in ComputeAsync()
95 OP_REQUIRES_ASYNC( in ComputeAsync()
243 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 | 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 | self_adjoint_eig_v2_op_gpu.cc | 51 OP_REQUIRES_ASYNC( in ComputeAsync()
56 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | check_numerics_op.cc | 143 OP_REQUIRES_ASYNC(context, stream != nullptr, in ComputeAsync()
167 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | functional_ops.cc | 140 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
227 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
244 OP_REQUIRES_ASYNC(ctx, TensorShapeUtils::IsScalar(branch_index.shape()), in ComputeAsync()
334 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
|
| 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 | 232 OP_REQUIRES_ASYNC( in ComputeAsync()
278 OP_REQUIRES_ASYNC(c, stream, errors::Internal("No GPU stream available."), in ComputeAsync()
296 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | where_op.cc | 294 OP_REQUIRES_ASYNC( in ComputeAsyncType()
339 OP_REQUIRES_ASYNC( in ComputeAsyncType()
|
| D | function_ops.cc | 243 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
305 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
|
| D | segment_reduction_ops.cc | 224 OP_REQUIRES_ASYNC( in ComputeAsync()
229 OP_REQUIRES_ASYNC( in ComputeAsync()
253 OP_REQUIRES_ASYNC( in ComputeAsync()
273 OP_REQUIRES_ASYNC(context, output_rows > 0, in ComputeAsync()
|
| D | sendrecv_ops.cc | 171 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | partitioned_function_ops.cc | 82 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
|
| D | qr_op_impl.h | 149 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| D | svd_op_gpu.cu.cc | 299 OP_REQUIRES_ASYNC( in ComputeAsync()
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| D | bigtable_kernels.cc | 196 OP_REQUIRES_ASYNC( in ComputeAsync()
202 OP_REQUIRES_ASYNC(ctx, columns_tensor->dims() == 1, in ComputeAsync()
205 OP_REQUIRES_ASYNC( in ComputeAsync()
236 OP_REQUIRES_ASYNC(ctx, timestamp_int >= -1, in ComputeAsync()
282 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/compiler/tf2xla/kernels/ |
| D | function_ops.cc | 68 OP_REQUIRES_ASYNC(ctx, lib != nullptr, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 729 OP_REQUIRES_ASYNC( in ComputeAsync()
736 OP_REQUIRES_ASYNC( in ComputeAsync()
1022 OP_REQUIRES_ASYNC( in ComputeAsync()
1030 OP_REQUIRES_ASYNC( in ComputeAsync()
|