Home
last modified time | relevance | path

Searched refs:FastBoundsCheck (Results 1 – 25 of 74) 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()
272 if (!FastBoundsCheck(k, lhs_right)) { in Compute()
275 if (!FastBoundsCheck(m, out.dimension(0))) { in Compute()
[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.cc89 if (!FastBoundsCheck(pad_top, window_rows)) { in CheckPaddingSize()
95 if (!FastBoundsCheck(pad_bottom, window_rows)) { in CheckPaddingSize()
101 if (!FastBoundsCheck(pad_left, window_cols)) { in CheckPaddingSize()
107 if (!FastBoundsCheck(pad_right, window_cols)) { in CheckPaddingSize()
326 FastBoundsCheck(input_pad_top, std::numeric_limits<int>::max()) && in Compute()
327 FastBoundsCheck(input_pad_bottom, std::numeric_limits<int>::max()) && in Compute()
328 FastBoundsCheck(input_pad_left, std::numeric_limits<int>::max()) && in Compute()
329 FastBoundsCheck(input_pad_right, std::numeric_limits<int>::max()); in Compute()
633 FastBoundsCheck(input_pad_top, std::numeric_limits<int>::max()) && in Compute()
634 FastBoundsCheck(input_pad_bottom, std::numeric_limits<int>::max()) && in Compute()
[all …]
Dscatter_functor.h160 if (!FastBoundsCheck(index, limit)) {
193 if (!FastBoundsCheck(index, limit)) return i;
250 if (!FastBoundsCheck(index, limit)) return i;
286 if (!FastBoundsCheck(index, limit)) return i;
298 if (!FastBoundsCheck(index, limit)) return i;
335 if (!FastBoundsCheck(index, limit)) return i;
360 if (!FastBoundsCheck(index, limit)) return i;
395 if (!FastBoundsCheck(index, limit)) return i;
Dctc_loss_op.cc111 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
154 OP_REQUIRES(ctx, FastBoundsCheck(batch_indices, batch_size), in Compute()
279 FastBoundsCheck(max_time_raw, std::numeric_limits<int>::max()), in Compute()
282 ctx, FastBoundsCheck(batch_size_raw, std::numeric_limits<int>::max()), in Compute()
285 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
Dsparse_tensor_dense_matmul_op_gpu.cu.cc42 if (!FastBoundsCheck(i, m)) { in SparseTensorDenseMatMulKernel()
47 if (!FastBoundsCheck(k, n)) { in SparseTensorDenseMatMulKernel()
Dsubstr_op.cc94 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
121 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
191 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
248 context, FastBoundsCheck(byte_pos, in.size() + 1), in Compute()
Dconv_ops.cc168 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
206 if (!FastBoundsCheck(explicit_padding, std::numeric_limits<int>::max())) { in operator ()()
414 FastBoundsCheck(filter.dim_size(i), std::numeric_limits<int>::max()), in ComputeConv2DDimension()
422 TF_REQUIRES(FastBoundsCheck(in_depth_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
424 TF_REQUIRES(FastBoundsCheck(patch_depth_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
439 TF_REQUIRES(FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
447 TF_REQUIRES(FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
454 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 …]
Done_hot_op.h99 if (FastBoundsCheck(depth, depth_size)) {
111 if (FastBoundsCheck(depth, depth_size)) {
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()
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()
Disotonic_regression_op.cc182 tensorflow::FastBoundsCheck(input.dimensions()[0], int_max) && in Compute()
183 tensorflow::FastBoundsCheck(input.dimensions()[1], int_max), in Compute()
Dsegment_reduction_ops_impl.h162 context, FastBoundsCheck(out_index, output_rows), in Compute()
336 OP_REQUIRES(ctx, FastBoundsCheck(j, num_segments),
541 context, FastBoundsCheck(out_index, output_rows),
646 if (!FastBoundsCheck(index##n, input_flat.dimension(0))) return (i);
887 context, FastBoundsCheck(idx, num_segments),
906 OP_REQUIRES(context, FastBoundsCheck(output_idx, M),
912 context, FastBoundsCheck(idx, num_segments),
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()
Dsearchsorted_op.cc104 FastBoundsCheck(sorted_inputs_t.dim_size(1), in Compute()
145 FastBoundsCheck(sorted_inputs_t.dim_size(1), in Compute()
Dsparse_xent_op.h74 if (!FastBoundsCheck(label, max_depth_)) { in operator()
114 if (!FastBoundsCheck(label, max_depth_)) { in operator()
Din_topk_op.h74 bool cannot_say = !FastBoundsCheck(target, num_classes) ||
/external/tensorflow/tensorflow/core/util/
Dimage_resizer_state.h102 FastBoundsCheck(input.dim_size(1), std::numeric_limits<int32>::max()) && in ValidateAndCalculateOutputSize()
103 FastBoundsCheck(input.dim_size(2), in ValidateAndCalculateOutputSize()
211 FastBoundsCheck(original_height, std::numeric_limits<int32>::max()) && in ValidateAndCreateOutput()
212 FastBoundsCheck(original_width, std::numeric_limits<int32>::max()), in ValidateAndCreateOutput()
/external/tensorflow/tensorflow/core/kernels/image/
Dencode_png_op.cc59 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()), in Compute()
68 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/compiler/tf2xla/kernels/
Dtranspose_op.cc147 FastBoundsCheck(ctx->InputShape(0).num_elements(), in InvertPermutation()
171 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.cc349 FastBoundsCheck(out_planes, std::numeric_limits<int>::max()), in Init()
354 FastBoundsCheck(out_height, std::numeric_limits<int>::max()), in Init()
358 FastBoundsCheck(out_width, std::numeric_limits<int>::max()), in Init()

123