| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_tensor_dense_matmul_op.cc | 103 (FastBoundsCheck(inner_left, int32max) &&
104 FastBoundsCheck(inner_right, int32max) &&
105 FastBoundsCheck(outer_left, int32max) &&
106 FastBoundsCheck(outer_right, int32max) &&
107 FastBoundsCheck(b->NumElements(), int32max) &&
108 FastBoundsCheck(outer_left * outer_right, int32max) &&
109 FastBoundsCheck(a_values->NumElements(), int32max)),
111 OP_REQUIRES(ctx, FastBoundsCheck(nnz * outer_right, int32max),
270 if (!FastBoundsCheck(k, lhs_right)) {
273 if (!FastBoundsCheck(m, out.dimension(0)))
[all...] |
| bounds_check.h | 30 EIGEN_ALWAYS_INLINE EIGEN_DEVICE_FUNC bool FastBoundsCheck(const Ta index,
33 "FastBoundsCheck can only be used on integer types.");
|
| image_resizer_state.h | 75 FastBoundsCheck(input.dim_size(1), std::numeric_limits<int32>::max()) &&
76 FastBoundsCheck(input.dim_size(2),
161 FastBoundsCheck(original_height, std::numeric_limits<int32>::max()) &&
162 FastBoundsCheck(original_width, std::numeric_limits<int32>::max()),
|
| dynamic_partition_op.cc | 61 OP_REQUIRES(c, FastBoundsCheck(p, num_partitions_),
115 c, FastBoundsCheck(p, num_partitions_),
118 OP_REQUIRES(c, FastBoundsCheck(oi, out_vec[p].size()),
143 c, FastBoundsCheck(p, num_partitions_),
148 OP_REQUIRES(c, FastBoundsCheck(oi, out_flat[p].dimension(0)),
|
| encode_png_op.cc | 59 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()),
68 OP_REQUIRES(context, FastBoundsCheck(width * channels, max_row_width),
|
| sparse_tensor_dense_matmul_op_gpu.cu.cc | 44 if (!FastBoundsCheck(i, m)) {
49 if (!FastBoundsCheck(k, n)) {
|
| encode_wav_op.cc | 47 FastBoundsCheck(audio.NumElements(), std::numeric_limits<int32>::max()),
|
| substr_op.cc | 69 context, FastBoundsCheck(pos, in.size() + 1),
83 context, FastBoundsCheck(pos, in.size() + 1),
149 context, FastBoundsCheck(pos, input_bcast(i).size() + 1),
200 OP_REQUIRES(context, FastBoundsCheck(pos, in.size() + 1),
|
| scatter_functor.h | 148 if (!FastBoundsCheck(index, limit)) return i;
172 if (!FastBoundsCheck(index, limit)) return i;
197 if (!FastBoundsCheck(index, limit)) return i;
208 if (!FastBoundsCheck(index, limit)) return i;
234 if (!FastBoundsCheck(index, limit)) return i;
|
| mkl_pooling_ops_common.cc | 115 FastBoundsCheck(out_height, std::numeric_limits<int>::max()),
119 FastBoundsCheck(out_width, std::numeric_limits<int>::max()),
|
| shape_ops.h | 65 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()),
95 ctx, FastBoundsCheck(dim_size, std::numeric_limits<int32>::max()),
138 ctx, FastBoundsCheck(size, std::numeric_limits<int32>::max()),
|
| conv_ops_using_gemm.cc | 473 FastBoundsCheck(filter.dim_size(i), std::numeric_limits<int>::max()),
493 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()),
503 FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()),
511 FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()),
|
| gather_functor_gpu.cu.h | 57 if (!FastBoundsCheck(gather_i, gather_dim_size)) {
|
| in_topk_op.cc | 77 OP_REQUIRES(context, FastBoundsCheck(target, num_classes),
|
| ctc_loss_op.cc | 73 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()),
114 OP_REQUIRES(ctx, FastBoundsCheck(batch_indices, batch_size),
|
| sparse_xent_op.h | 73 if (!FastBoundsCheck(label, max_depth_)) {
113 if (!FastBoundsCheck(label, max_depth_)) {
|
| depthwise_conv_op.cc | 325 FastBoundsCheck(input_rows_raw, std::numeric_limits<int32>::max()),
333 FastBoundsCheck(input_cols_raw, std::numeric_limits<int32>::max()),
353 FastBoundsCheck(out_shape.num_elements(),
|
| split_op.cc | 146 FastBoundsCheck(input.NumElements(),
251 FastBoundsCheck(input.NumElements(), std::numeric_limits<int32>::max()),
310 FastBoundsCheck(input.NumElements(),
|
| encode_jpeg_op.cc | 85 FastBoundsCheck(image.NumElements(), std::numeric_limits<int32>::max()),
|
| gather_nd_op_cpu_impl.h | 61 out_of_bounds |= !FastBoundsCheck(ix_i, Tparams_.dimension(i));
|
| gather_nd_op_gpu.cu.cc | 45 out_of_bounds |= !FastBoundsCheck(index_j, batch_indices[j]);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| shape_util.cc | 31 if (!FastBoundsCheck(dim_size, std::numeric_limits<int32>::max())) {
|
| transpose_op.cc | 109 FastBoundsCheck(ctx->InputShape(0).num_elements(),
123 OP_REQUIRES(ctx, FastBoundsCheck(d, size),
|
| scan_ops.cc | 64 ctx, FastBoundsCheck(axis, input_shape.dims()),
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| masked_matmul_ops.cc | 121 CHECK(FastBoundsCheck(a_index, a_dim_0))
128 CHECK(FastBoundsCheck(b_index, b_dim_1))
|
