- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getEndMask (0.33 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
loc, revised_output_type, original_input, shape); // Replace the original strided_slice. uint64_t revised_begin_mask = strided_slice_op.getBeginMask(); uint64_t revised_end_mask = strided_slice_op.getEndMask(); // Since we expand the dims, we need to apply them to the begin_mask & // end_mask. revised_begin_mask |= strided_slice_op.getNewAxisMask(); revised_end_mask |= strided_slice_op.getNewAxisMask();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// All masks are `0` except `shrink_axis_mask` which is equal to `1` (slicing // scalar value from input vector). if (slice_op.getBeginMask() != 0 || slice_op.getEllipsisMask() != 0 || slice_op.getEndMask() != 0 || slice_op.getNewAxisMask() != 0 || slice_op.getShrinkAxisMask() != 1) return {}; // Returns a value if the `value` is defined by a ConstOp with a single
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
strided_slice_op.getEnd().getLoc(), new_added_value); if (strided_slice_op.getBeginMask() != 0) return failure(); if (strided_slice_op.getEndMask() != 0) return failure(); if (strided_slice_op.getEllipsisMask() != 0) return failure(); mlir::TFL::StridedSliceOp new_strided_slice_op = rewriter.create<TFL::StridedSliceOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
if (num_input_dims + num_added_axis > 5) return failure(); return success(); } OpFoldResult StridedSliceOp::fold(FoldAdaptor) { // Currently only support all masks being 0. if (getBeginMask() != 0 || getEndMask() != 0 || getEllipsisMask() != 0 || getNewAxisMask() != 0 || getShrinkAxisMask() != 0) return {}; auto input_type = getInput().getType().dyn_cast_or_null<RankedTensorType>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// verifier. int64_t slicing_dim_size = mlir::cast<RankedTensorType>(op.getBegin().getType()).getDimSize(0); uint64_t begin_mask = op.getBeginMask(); uint64_t end_mask = op.getEndMask(); const int input_rank = input_shape.size(); for (int d = 0; d < input_rank; ++d) { // Each dimension is either sliced fully or has size of one.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)