Searched refs:padded_end (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 63 std::vector<int64> padded_end(n_dims); in SliceInMinorDims() local
64 std::copy(major_dims.begin(), major_dims.end(), padded_end.begin()); in SliceInMinorDims()
65 std::copy(end.begin(), end.end(), padded_end.begin() + major_dims.size()); in SliceInMinorDims()
68 return Slice(x, padded_start, padded_end, strides); in SliceInMinorDims()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 652 SmallVector<int32_t, 4> padded_end; in RewriteEllipsisMask() local
660 padded_end.push_back(end_dense_elem_attr.getValue<int32_t>(index)); in RewriteEllipsisMask()
677 padded_end.push_back(0); in RewriteEllipsisMask()
687 padded_end.push_back(end_dense_elem_attr.getValue<int32_t>(index)); in RewriteEllipsisMask()
707 auto end_attr = DenseElementsAttr::get<int32_t>(type, padded_end); in RewriteEllipsisMask()
791 SmallVector<int32_t, 4> padded_begin, padded_end, padded_strides; in matchAndRewrite() local
804 PadStridedSliceAttributeArray(end_elem_attr, end, padded_end, padding_end, in matchAndRewrite()
809 if (begin == padded_begin && end == padded_end && in matchAndRewrite()
823 DenseElementsAttr::get<int32_t>(begin_end_type, padded_end)); in matchAndRewrite()
|