Home
last modified time | relevance | path

Searched refs:FastBoundsCheck (Results 1 – 25 of 75) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensor_dense_matmul_op.cc104 (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()
278 if (!FastBoundsCheck(k, lhs_right)) { in SparseTensorDenseMatMulImpl()
281 if (!FastBoundsCheck(m, out.dimension(0))) { in SparseTensorDenseMatMulImpl()
[all …]
Ddynamic_partition_op.cc61 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()
Dpooling_ops_common.cc90 if (!FastBoundsCheck(pad_top, window_rows)) { in CheckPaddingSize()
96 if (!FastBoundsCheck(pad_bottom, window_rows)) { in CheckPaddingSize()
102 if (!FastBoundsCheck(pad_left, window_cols)) { in CheckPaddingSize()
108 if (!FastBoundsCheck(pad_right, window_cols)) { in CheckPaddingSize()
329 FastBoundsCheck(input_pad_top, std::numeric_limits<int>::max()) && in Compute()
330 FastBoundsCheck(input_pad_bottom, std::numeric_limits<int>::max()) && in Compute()
331 FastBoundsCheck(input_pad_left, std::numeric_limits<int>::max()) && in Compute()
332 FastBoundsCheck(input_pad_right, std::numeric_limits<int>::max()); in Compute()
642 FastBoundsCheck(input_pad_top, std::numeric_limits<int>::max()) && in Compute()
643 FastBoundsCheck(input_pad_bottom, std::numeric_limits<int>::max()) && in Compute()
[all …]
Dscatter_functor.h161 if (!FastBoundsCheck(index, limit)) {
194 if (!FastBoundsCheck(index, limit)) return i;
252 if (!FastBoundsCheck(index, limit)) return i;
288 if (!FastBoundsCheck(index, limit)) return i;
300 if (!FastBoundsCheck(index, limit)) return i;
337 if (!FastBoundsCheck(index, limit)) return i;
362 if (!FastBoundsCheck(index, limit)) return i;
397 if (!FastBoundsCheck(index, limit)) return i;
Dctc_loss_op.cc120 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
163 OP_REQUIRES(ctx, FastBoundsCheck(batch_indices, batch_size), in Compute()
288 FastBoundsCheck(max_time_raw, std::numeric_limits<int>::max()), in Compute()
291 ctx, FastBoundsCheck(batch_size_raw, std::numeric_limits<int>::max()), in Compute()
294 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
Dsubstr_op.cc94 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
121 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
192 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
249 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
Dconv_ops.cc267 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
351 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
559 FastBoundsCheck(filter.dim_size(i), std::numeric_limits<int>::max()), in ComputeConv2DDimension()
567 TF_REQUIRES(FastBoundsCheck(in_depth_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
569 TF_REQUIRES(FastBoundsCheck(patch_depth_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
587 TF_REQUIRES(FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
595 TF_REQUIRES(FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
602 TF_REQUIRES(FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
968 FastBoundsCheck(input_pad_top, std::numeric_limits<int>::max()) && in operator ()()
969 FastBoundsCheck(input_pad_bottom, std::numeric_limits<int>::max()) && in operator ()()
[all …]
Done_hot_op.h99 if (FastBoundsCheck(depth, depth_size)) {
111 if (FastBoundsCheck(depth, depth_size)) {
Dconv_ops_using_gemm.cc475 FastBoundsCheck(filter.dim_size(i), std::numeric_limits<int>::max()), in Compute()
495 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in Compute()
505 FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in Compute()
513 FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()), in Compute()
Dshape_ops.h57 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()), in Compute()
86 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()), in Compute()
127 ctx, FastBoundsCheck(size, std::numeric_limits<int32>::max()), in Compute()
Disotonic_regression_op.cc183 tensorflow::FastBoundsCheck(input.dimensions()[0], int_max) && in Compute()
184 tensorflow::FastBoundsCheck(input.dimensions()[1], int_max), in Compute()
Dsparse_tensor_dense_matmul_op_gpu.cu.cc58 if (!FastBoundsCheck(i, m)) { in SparseTensorDenseMatMulKernel()
63 if (!FastBoundsCheck(k, n)) { in SparseTensorDenseMatMulKernel()
Dencode_wav_op.cc47 FastBoundsCheck(audio.NumElements(), std::numeric_limits<int32>::max()), in Compute()
Dtranspose_op.cc54 FastBoundsCheck(Tin.size(), std::numeric_limits<int32>::max()), in Compute()
65 OP_REQUIRES(context, FastBoundsCheck(d, N), in Compute()
Dctc_decoder_ops.cc211 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
229 OP_REQUIRES(ctx, FastBoundsCheck(blank_index, num_classes), in Compute()
319 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
Dsparse_xent_op.h74 if (!FastBoundsCheck(label, max_depth_)) { in operator()
114 if (!FastBoundsCheck(label, max_depth_)) { in operator()
Dsegment_reduction_ops_impl.h160 context, FastBoundsCheck(out_index, output_rows), in Compute()
387 OP_REQUIRES(ctx, FastBoundsCheck(j, num_segments),
627 context, FastBoundsCheck(out_index, output_rows),
738 if (!FastBoundsCheck(index##n, input_flat.dimension(0))) return (i);
1057 context, FastBoundsCheck(idx, num_segments),
1089 OP_REQUIRES(context, FastBoundsCheck(output_idx, M),
1095 context, FastBoundsCheck(idx, num_segments),
/external/tensorflow/tensorflow/core/util/
Dimage_resizer_state.h102 FastBoundsCheck(input_shape.dim_size(1), in ValidateAndCalculateOutputSize()
104 FastBoundsCheck(input_shape.dim_size(2), in ValidateAndCalculateOutputSize()
225 FastBoundsCheck(original_height, std::numeric_limits<int32>::max()) && in ValidateAndCreateOutput()
226 FastBoundsCheck(original_width, std::numeric_limits<int32>::max()), in ValidateAndCreateOutput()
/external/tensorflow/tensorflow/core/kernels/image/
Dencode_png_op.cc61 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()), in Compute()
70 OP_REQUIRES(context, FastBoundsCheck(width * channels, max_row_width), in Compute()
Dencode_jpeg_op.cc85 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()), in Compute()
151 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()), in Compute()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-068.md33 FastBoundsCheck(input_backprop_index - in_start, in_end - in_start);
38 Whereas accesses to `input_backprop_flat` are guarded by `FastBoundsCheck`, the
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtranspose_op.cc147 FastBoundsCheck(ctx->InputShape(0).num_elements(), in InvertPermutation()
172 OP_REQUIRES(ctx, FastBoundsCheck(d, size), in InvertPermutation()
Dshape_util.cc31 if (!FastBoundsCheck(dim_size, std::numeric_limits<int32>::max())) { in TensorShapeToConstant()
/external/tensorflow/tensorflow/core/framework/
Dbounds_check.h30 EIGEN_ALWAYS_INLINE EIGEN_DEVICE_FUNC bool FastBoundsCheck(const Ta index, in FastBoundsCheck() function
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc353 FastBoundsCheck(out_planes, std::numeric_limits<int>::max()), in Init()
358 FastBoundsCheck(out_height, std::numeric_limits<int>::max()), in Init()
362 FastBoundsCheck(out_width, std::numeric_limits<int>::max()), in Init()

123