| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 39 int32 ellipsis_mask; member
107 if ((1 << i) & sparse.ellipsis_mask) { in BuildDenseSpec()
172 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp() argument
204 if (ellipsis_mask && ((ellipsis_mask & (ellipsis_mask - 1)) != 0)) { in ValidateStridedSliceOp()
223 ellipsis_mask, in ValidateStridedSliceOp()
231 if ((1 << i) & ellipsis_mask) { in ValidateStridedSliceOp()
237 sparse_spec.ellipsis_mask |= (1 << sparse_spec.dims); in ValidateStridedSliceOp()
420 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp() argument
430 end_mask_spec, ellipsis_mask, new_axis_mask, shrink_axis_mask, in ValidateStridedSliceOp()
|
| D | strided_slice_op.h | 67 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask,
79 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | fuse-tftext.mlir | 29 …%25 = "tf.StridedSlice"(%24#0, %15, %5, %16) {begin_mask = 1 : i64, device = "", ellipsis_mask = 0…
31 …%27 = "tf.StridedSlice"(%24#0, %16, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
34 …%30 = "tf.StridedSlice"(%29#0, %5, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0…
41 …%37 = "tf.StridedSlice"(%36#1, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
46 …%42 = "tf.StridedSlice"(%36#1, %16, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
47 …%43 = "tf.StridedSlice"(%36#1, %15, %5, %16) {begin_mask = 1 : i64, device = "", ellipsis_mask = 0…
54 …%50 = "tf.StridedSlice"(%49, %5, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 :…
56 …%52 = "tf.StridedSlice"(%51, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 …
63 …%59 = "tf.StridedSlice"(%58, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 …
65 …%61 = "tf.StridedSlice"(%36#4, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
[all …]
|
| D | prepare-tf.mlir | 459 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
464 …dSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 3 : i64, ellipsis_mask = 0 : i64, end…
472 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst_0, %cst_1) {begin_mask = 6 : i64, ellipsis_mask = 1 : i64…
478 …lice"(%arg0, %[[CST]], %[[CST_0]], %[[CST_1]]) {begin_mask = 7 : i64, ellipsis_mask = 0 : i64, end…
485 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 2 : i64, …
490 …dSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 6 : i64, ellipsis_mask = 0 : i64, end…
497 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
502 …dSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, end…
509 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask = 0 : i64,…
514 …// CHECK: %0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask …
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 87 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceOp()
106 ellipsis_mask, new_axis_mask, shrink_axis_mask, in Compute()
185 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
195 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceGradOp()
232 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
279 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
289 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceAssignOp()
342 old_lhs->shape(), begin_mask, end_mask, ellipsis_mask, in Compute()
391 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
| D | strided_slice_op_test.cc | 111 const int32 ellipsis_mask = 0; in BM_ValidateStridedSliceOp() local
116 &begin, &end, strides, input_shape, begin_mask, end_mask, ellipsis_mask, in BM_ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 568 int32 end_mask, int32 ellipsis_mask, in StridedSliceGrad() argument
588 {"ellipsis_mask", ellipsis_mask}, in StridedSliceGrad()
610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() argument
631 {"ellipsis_mask", ellipsis_mask}, in StridedSliceGradGrad()
665 ellipsis_mask = 0; in TEST() local
668 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
679 begin_mask, end_mask, ellipsis_mask, in TEST()
692 ellipsis_mask = 0; in TEST() local
695 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
706 begin_mask, end_mask, ellipsis_mask, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_StridedSlice.pbtxt | 46 name: "ellipsis_mask"
50 If `ellipsis_mask == 0`, then an implicit ellipsis mask of `1 << (m+1)`
89 `ellipsis_mask`. For example, `foo[...]` is the identity slice.
128 ellipsis_mask = 1<<3 = 8
165 `ellipsis_mask must be a power of two (only one ellipsis)`
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 60 p->ellipsis_mask <<= pad_count; in StridedSlicePadIndices()
199 op_params.ellipsis_mask = 0; in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | StridedSlice.pbtxt | 52 name: "ellipsis_mask"
|
| D | ResourceStridedSliceAssign.pbtxt | 52 name: "ellipsis_mask"
|
| D | StridedSliceGrad.pbtxt | 56 name: "ellipsis_mask"
|
| D | TensorStridedSliceUpdate.pbtxt | 56 name: "ellipsis_mask"
|
| D | StridedSliceAssign.pbtxt | 58 name: "ellipsis_mask"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | StridedSlice.pbtxt | 52 name: "ellipsis_mask"
|
| D | ResourceStridedSliceAssign.pbtxt | 52 name: "ellipsis_mask"
|
| D | StridedSliceGrad.pbtxt | 56 name: "ellipsis_mask"
|
| D | TensorStridedSliceUpdate.pbtxt | 56 name: "ellipsis_mask"
|
| D | StridedSliceAssign.pbtxt | 58 name: "ellipsis_mask"
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice.cc | 76 op_params.ellipsis_mask = 0; in BuildStridedSliceParams()
169 TF_LITE_ENSURE_MSG(context, op_context.params->ellipsis_mask == 0, in Prepare()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 246 if (slice_op.begin_mask() != 0 || slice_op.ellipsis_mask() != 0 || in fold()
1628 uint32_t ellipsis_mask = op.ellipsis_mask(); in VerifyStridedSliceBase() local
1629 if (ellipsis_mask != 0 && !llvm::isPowerOf2_32(ellipsis_mask)) in VerifyStridedSliceBase()
1676 int32_t begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; member
1718 if (IsSet(sparse.ellipsis_mask, sparse_index)) ellipsis_seen = true; in BuildDenseSliceSpec()
1724 if (IsSet(sparse.ellipsis_mask, sparse_index)) { in BuildDenseSliceSpec()
1827 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in CalculateSlicedShapeFromSparseIndices() argument
1833 ellipsis_mask, new_axis_mask, shrink_axis_mask, in CalculateSlicedShapeFromSparseIndices()
1839 if (sparse.ellipsis_mask == 0) { in CalculateSlicedShapeFromSparseIndices()
1840 Set(sparse.ellipsis_mask, sparse.dims); in CalculateSlicedShapeFromSparseIndices()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 1547 ellipsis_mask = 0
1570 ellipsis_mask |= (1 << index)
1590 advanced_indices.append((index, s, ellipsis_mask != 0))
1624 ellipsis_mask=ellipsis_mask,
1641 ellipsis_mask=ellipsis_mask,
1665 ellipsis_mask=ellipsis_mask,
1743 ellipsis_mask=ellipsis_mask,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 586 strided_slice_op.ellipsis_mask()), in RewriteNewAxisMask()
597 uint64_t ellipsis_mask = strided_slice_op.ellipsis_mask(); in RewriteEllipsisMask() local
601 shrink_axis_mask &= ~ellipsis_mask; in RewriteEllipsisMask()
658 while (((ellipsis_mask >> index) & 1) == 0) { in RewriteEllipsisMask()
755 if (strided_slice_op.ellipsis_mask() != 0) { in matchAndRewrite()
838 strided_slice_op.ellipsis_mask()), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 1113 int ellipsis_mask;
1128 ellipsis_mask = other.ellipsis_mask;
1159 ellipsis_mask =
1160 toco::port::ReverseBits32(static_cast<uint32>(ellipsis_mask)) >>
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 165 int64_t ellipsis_mask, int64_t new_axis_mask, in CreateStridedSliceOp() argument
177 builder->getI64IntegerAttr(ellipsis_mask), in CreateStridedSliceOp()
778 int64_t ellipsis_mask = 0; in ConvertKerasLSTMLayer() local
796 end, strides, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in ConvertKerasLSTMLayer()
|