| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op.cc | 104 (FastBoundsCheck(inner_left, int32max) && in Compute()
105 FastBoundsCheck(inner_right, int32max) && in Compute()
106 FastBoundsCheck(outer_left, int32max) && in Compute()
107 FastBoundsCheck(outer_right, int32max) && in Compute()
108 FastBoundsCheck(b->NumElements(), int32max) && in Compute()
109 FastBoundsCheck(outer_left * outer_right, int32max) && in Compute()
110 FastBoundsCheck(a_values->NumElements(), int32max)), in Compute()
112 OP_REQUIRES(ctx, FastBoundsCheck(nnz * outer_right, int32max), in Compute()
272 if (!FastBoundsCheck(k, lhs_right)) { in Compute()
275 if (!FastBoundsCheck(m, out.dimension(0))) { in Compute()
[all …]
|
| D | dynamic_partition_op.cc | 61 OP_REQUIRES(c, FastBoundsCheck(p, num_partitions_), in ValidateAndAllocateOutputs()
115 c, FastBoundsCheck(p, num_partitions_), in Compute()
118 OP_REQUIRES(c, FastBoundsCheck(oi, out_vec[p].size()), in Compute()
143 c, FastBoundsCheck(p, num_partitions_), in Compute()
148 OP_REQUIRES(c, FastBoundsCheck(oi, out_flat[p].dimension(0)), in Compute()
|
| D | scatter_functor.h | 222 if (!FastBoundsCheck(index, limit)) {
255 if (!FastBoundsCheck(index, limit)) return i;
312 if (!FastBoundsCheck(index, limit)) return i;
346 if (!FastBoundsCheck(index, limit)) return i;
372 if (!FastBoundsCheck(index, limit)) return i;
384 if (!FastBoundsCheck(index, limit)) return i;
410 if (!FastBoundsCheck(index, limit)) return i;
442 if (!FastBoundsCheck(index, limit)) return i;
467 if (!FastBoundsCheck(index, limit)) return i;
501 if (!FastBoundsCheck(index, limit)) return i;
[all …]
|
| D | image_resizer_state.h | 102 FastBoundsCheck(input.dim_size(1), std::numeric_limits<int32>::max()) && in ValidateAndCalculateOutputSize()
103 FastBoundsCheck(input.dim_size(2), in ValidateAndCalculateOutputSize()
197 FastBoundsCheck(original_height, std::numeric_limits<int32>::max()) && in ValidateAndCreateOutput()
198 FastBoundsCheck(original_width, std::numeric_limits<int32>::max()), in ValidateAndCreateOutput()
|
| D | ctc_loss_op.cc | 108 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
151 OP_REQUIRES(ctx, FastBoundsCheck(batch_indices, batch_size), in Compute()
276 FastBoundsCheck(max_time_raw, std::numeric_limits<int>::max()), in Compute()
279 ctx, FastBoundsCheck(batch_size_raw, std::numeric_limits<int>::max()), in Compute()
282 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
|
| D | encode_png_op.cc | 59 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()), in Compute()
68 OP_REQUIRES(context, FastBoundsCheck(width * channels, max_row_width), in Compute()
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 42 if (!FastBoundsCheck(i, m)) { in SparseTensorDenseMatMulKernel()
47 if (!FastBoundsCheck(k, n)) { in SparseTensorDenseMatMulKernel()
|
| D | conv_ops.cc | 167 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
205 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
413 FastBoundsCheck(filter.dim_size(i), std::numeric_limits<int>::max()), in ComputeConv2DDimension()
421 TF_REQUIRES(FastBoundsCheck(in_depth_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
423 TF_REQUIRES(FastBoundsCheck(patch_depth_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
438 TF_REQUIRES(FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
446 TF_REQUIRES(FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
453 TF_REQUIRES(FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
817 FastBoundsCheck(input_pad_top, std::numeric_limits<int>::max()) && in operator ()()
818 FastBoundsCheck(input_pad_bottom, std::numeric_limits<int>::max()) && in operator ()()
[all …]
|
| D | substr_op.cc | 89 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
116 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
196 FastBoundsCheck(byte_pos, input_bcast(i).size() + 1), in Compute()
261 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
|
| D | one_hot_op.h | 99 if (FastBoundsCheck(depth, depth_size)) {
111 if (FastBoundsCheck(depth, depth_size)) {
|
| D | conv_ops_using_gemm.cc | 473 FastBoundsCheck(filter.dim_size(i), std::numeric_limits<int>::max()), in Compute()
493 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in Compute()
503 FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in Compute()
511 FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()), in Compute()
|
| D | shape_ops.h | 55 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()), in Compute()
84 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()), in Compute()
125 ctx, FastBoundsCheck(size, std::numeric_limits<int32>::max()), in Compute()
|
| D | segment_reduction_ops_impl.h | 162 context, FastBoundsCheck(out_index, output_rows), in Compute()
334 OP_REQUIRES(ctx, FastBoundsCheck(j, num_segments),
528 context, FastBoundsCheck(out_index, output_rows),
578 if (!FastBoundsCheck(index##n, input_flat.dimension(0))) return (i);
819 context, FastBoundsCheck(idx, num_segments),
838 OP_REQUIRES(context, FastBoundsCheck(output_idx, M),
844 context, FastBoundsCheck(idx, num_segments),
|
| D | encode_jpeg_op.cc | 85 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()), in Compute()
151 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()), in Compute()
|
| D | encode_wav_op.cc | 47 FastBoundsCheck(audio.NumElements(), std::numeric_limits<int32>::max()), in Compute()
|
| D | searchsorted_op.cc | 104 FastBoundsCheck(sorted_inputs_t.dim_size(1), in Compute()
145 FastBoundsCheck(sorted_inputs_t.dim_size(1), in Compute()
|
| D | sparse_xent_op.h | 74 if (!FastBoundsCheck(label, max_depth_)) { in operator()
114 if (!FastBoundsCheck(label, max_depth_)) { in operator()
|
| D | split_op.cc | 223 FastBoundsCheck(input.NumElements(), in Compute()
292 FastBoundsCheck(input.NumElements(), std::numeric_limits<int32>::max()), in Compute()
351 FastBoundsCheck(input.NumElements(), in Compute()
|
| D | depthwise_conv_op.cc | 343 FastBoundsCheck(input_rows_raw, std::numeric_limits<int32>::max()), in Compute()
351 FastBoundsCheck(input_cols_raw, std::numeric_limits<int32>::max()), in Compute()
371 FastBoundsCheck(out_shape.num_elements(), in Compute()
|
| D | mkl_pooling_ops_common.cc | 347 FastBoundsCheck(out_planes, std::numeric_limits<int>::max()), in Init()
352 FastBoundsCheck(out_height, std::numeric_limits<int>::max()), in Init()
356 FastBoundsCheck(out_width, std::numeric_limits<int>::max()), in Init()
|
| D | scatter_nd_op_cpu_impl.h | 120 out_of_bounds |= !FastBoundsCheck(ix_d, output_shape_prefix[dim]);
204 out_of_bounds |= !FastBoundsCheck(ix_d, output_shape_prefix[dim]);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | transpose_op.cc | 126 FastBoundsCheck(ctx->InputShape(0).num_elements(), in Compile()
149 OP_REQUIRES(ctx, FastBoundsCheck(d, size), in Compile()
|
| D | shape_util.cc | 31 if (!FastBoundsCheck(dim_size, std::numeric_limits<int32>::max())) { in TensorShapeToConstant()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | bounds_check.h | 30 EIGEN_ALWAYS_INLINE EIGEN_DEVICE_FUNC bool FastBoundsCheck(const Ta index, in FastBoundsCheck() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 62 if (!FastBoundsCheck(dimension, std::numeric_limits<int32>::max())) { in TryToInferTensorOutputFromInputShapes()
100 if (!FastBoundsCheck(size, std::numeric_limits<int32>::max())) { in TryToInferTensorOutputFromInputShapes()
|