Home
last modified time | relevance | path

Searched refs:upper_bounds (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_with_default_op.py86 upper_bounds = None
111 upper_bounds = math_ops.cast(row_lengths, indices.dtype)
124 upper_bounds = params_shape[num_batch_dimensions]
135 is_out_of_bounds = (indices < 0) | (indices > upper_bounds)
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_affine.cc34 OpBuilder& builder, Location location, ArrayRef<int64_t> upper_bounds, in BuildBoundedAffineLoopNest() argument
36 SmallVector<int64_t, 3> lower_bounds(upper_bounds.size(), /*Value=*/0); in BuildBoundedAffineLoopNest()
37 SmallVector<int64_t, 3> steps(upper_bounds.size(), /*Value=*/1); in BuildBoundedAffineLoopNest()
38 buildAffineLoopNest(builder, location, lower_bounds, upper_bounds, steps, in BuildBoundedAffineLoopNest()
/external/llvm-project/flang/runtime/
DISO_Fortran_binding.cpp39 const CFI_index_t upper_bounds[], std::size_t elem_len) { in CFI_allocate() argument
73 CFI_index_t ub{upper_bounds[j]}; in CFI_allocate()
275 const CFI_index_t lower_bounds[], const CFI_index_t upper_bounds[], in CFI_section() argument
287 if (IsAssumedSize(source) && !upper_bounds) { in CFI_section()
308 const CFI_index_t ub{upper_bounds ? upper_bounds[j] : srcUB}; in CFI_section()
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter_transforms.cc67 absl::Span<const int64_t> upper_bounds, OpBuilder builder) { in CreateNestedSimpleLoops() argument
69 loops.reserve(upper_bounds.size()); in CreateNestedSimpleLoops()
70 for (int64_t dim : upper_bounds) { in CreateNestedSimpleLoops()
Dconv_emitter_transforms.h40 absl::Span<const int64_t> upper_bounds, mlir::OpBuilder builder);
/external/llvm-project/flang/include/flang/
DISO_Fortran_binding.h166 const CFI_index_t upper_bounds[], size_t elem_len);
172 const CFI_index_t lower_bounds[], const CFI_index_t upper_bounds[],
/external/llvm-project/flang/unittests/Evaluate/
DISO-Fortran-binding.cpp287 const CFI_index_t lower_bounds[], const CFI_index_t upper_bounds[], in check_CFI_allocate() argument
300 int retCode{CFI_allocate(dv, lower_bounds, upper_bounds, elem_len)}; in check_CFI_allocate()
307 MATCH(upper_bounds[i], dv->dim[i].extent + dv->dim[i].lower_bound - 1); in check_CFI_allocate()
349 if (rank > 0 && (lower_bounds == nullptr || upper_bounds == nullptr)) { in check_CFI_allocate()
/external/googletest/docs/reference/
Dmatchers.md124 …g. `Pointwise(Le(), upper_bounds)` verifies that each element in `argument` doesn't exceed the cor…
/external/llvm-project/polly/lib/External/isl/
Disl_ast_codegen.c643 static __isl_give isl_pw_aff_list *upper_bounds( in upper_bounds() function
1205 upper_list = upper_bounds(c_upper, depth, domain, build); in refine_generic_bounds()