- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getEllipsisMask (0.41 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
TF::StridedSliceOp strided_slice_op = llvm::cast<TF::StridedSliceOp>(op); uint64_t new_axis_mask = strided_slice_op.getNewAxisMask(); if (strided_slice_op.getEllipsisMask() != 0) { // Ellipsis mask should have been lowered-away prior to invoking this // function. op->emitError() << "encountered a logical error"; return failure(); }
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
Value tensor = shape_op.getInput(); // 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>( strided_slice_op.getLoc(), strided_slice_op.getOutput().getType(),
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
} // The kernel will reshape the input tensor with new axis, it only supports // this reshaped tensor up to 5D. uint32_t ellipsis_mask = op.getEllipsisMask(); uint32_t new_axis_mask = op.getNewAxisMask(); int num_added_axis = 0; for (int i = 0; i < 8; ++i) { if (!((1 << i) & ellipsis_mask) && ((1 << i) & new_axis_mask)) { num_added_axis++;
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
int last_dim = std::max(static_cast<int>(input_shape.size()) - 1, 0); // When begin/end values are dynamic, the ellipsis mask, if set, must refer // to the last dimension. int ellipsis_mask = op.getEllipsisMask(); if (!(ellipsis_mask == 0 || ellipsis_mask == (1 << last_dim))) return rewriter.notifyMatchFailure( op, "requires that ellipsis_mask, if set, refer to the last dimension of "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)