Home
last modified time | relevance | path

Searched refs:begin_mask (Results 1 – 25 of 91) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt1763 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1764 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1765 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1766 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1767 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1768 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1769 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1770 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1771 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf…
1772 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf…
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h62 p->begin_mask <<= pad_count; in StridedSlicePadIndices()
64 p->begin_mask |= (1 << pad_count) - 1; in StridedSlicePadIndices()
77 const auto begin_mask = params.begin_mask; in StartForAxis() local
88 if (begin_mask & 1 << axis) { in StartForAxis()
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
198 op_params.begin_mask = begin_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dstrided_slice_builder.cc51 int begin_mask = params->begin_mask; in PopulateSubGraph() local
55 begin_mask = begin_mask << (4 - original_mask_size); in PopulateSubGraph()
59 kScalarShape, reinterpret_cast<char*>(&begin_mask), sizeof(begin_mask)); in PopulateSubGraph()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dfuse-tftext.mlir29 …%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 …]
Dprepare-tf.mlir459 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
464 …SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 3 : i64, ellip…
472 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst_0, %cst_1) {begin_mask = 6 : i64, ellipsis_mask = 1 : i64…
478 …ICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST_0]], %[[CST_1]]) {begin_mask = 7 : i64, ellip…
485 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 2 : i64, …
490 …SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 6 : i64, ellip…
497 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
502 …SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 0 : i64, ellip…
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/util/
Dstrided_slice_op.cc38 const int32 begin_mask, end_mask; member
50 int32 begin_mask; member
89 dense->begin_mask = 0; in BuildDenseSpec()
117 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
142 if (sparse.begin_mask & (1 << i)) { in BuildDenseSpec()
143 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
288 {dense_spec.begin_mask & (1 << i), dense_spec.end_mask & (1 << i)}}; in ValidateStridedSliceOp()
309 (dense_spec.begin_mask & (1 << i)) && (dense_spec.end_mask & (1 << i)); in ValidateStridedSliceOp()
389 shape_spec->begin_dense_mask = dense_spec.begin_mask; in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc85 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceOp()
105 context->input(0).shape(), begin_mask, end_mask, in Compute()
184 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
193 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceGradOp()
232 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
278 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp
287 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceAssignOp()
342 old_lhs->shape(), begin_mask, end_mask, ellipsis_mask, in Compute()
390 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc109 const int32 begin_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/
Darray_grad_test.cc567 const Tensor& dy, int32 begin_mask, in StridedSliceGrad() argument
584 {"begin_mask", begin_mask}, in StridedSliceGrad()
610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() argument
627 {"begin_mask", begin_mask}, in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
667 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
679 begin_mask, end_mask, ellipsis_mask, in TEST()
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
694 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
706 begin_mask, end_mask, ellipsis_mask, in TEST()
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt9 `begin_mask` then `begin[k]` is ignored and the full range of the
31 name: "begin_mask"
42 analogous to `begin_mask`
102 the second to last. `begin_mask` controls whether to replace the explicitly
126 begin_mask = 1<<4 | 1<<5 = 48
155 The appropriate begin_mask bit is set to indicate the start range is the
160 receive 0, 0, and 1, respectively. The appropriate bits in `begin_mask` and
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc75 op_params.begin_mask = op_context->params->begin_mask; in BuildStridedSliceParams()
93 op_params.begin_mask |= (1 << i); in BuildStridedSliceParams()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_strided_slice_attributes.cc111 op->begin_mask = in Run()
112 PadAttributeArray(&start_array, begin_pad_values, op->begin_mask); in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc572 uint64_t revised_begin_mask = strided_slice_op.begin_mask(); in RewriteNewAxisMask()
645 int64_t begin_mask = strided_slice_op.begin_mask(); in RewriteEllipsisMask() local
662 if ((begin_mask >> index) & 1) revised_begin_mask |= (1 << new_index); in RewriteEllipsisMask()
690 if ((begin_mask >> index) & 1) revised_begin_mask |= (1 << new_index); in RewriteEllipsisMask()
799 int begin_mask = strided_slice_op.begin_mask(); in matchAndRewrite() local
803 padding_begin, &begin_mask); in matchAndRewrite()
811 begin_mask == strided_slice_op.begin_mask() && in matchAndRewrite()
835 rewriter.getIntegerAttr(attribute_type, begin_mask), in matchAndRewrite()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dstrided_slice_test.cc30 std::initializer_list<int> strides_data, int begin_mask, in StridedSliceOpModel() argument
40 builder_, begin_mask, end_mask, /*ellipsis_mask*/ 0, in StridedSliceOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc164 int64_t begin_mask, int64_t end_mask, in CreateStridedSliceOp() argument
175 builder->getI64IntegerAttr(begin_mask), in CreateStridedSliceOp()
779 int64_t begin_mask; in ConvertKerasLSTMLayer() local
783 begin_mask = 6; in ConvertKerasLSTMLayer()
788 begin_mask = 5; in ConvertKerasLSTMLayer()
796 end, strides, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in ConvertKerasLSTMLayer()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice.cc74 op_params.begin_mask = op_context->params->begin_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc586 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
588 GetNodeAttr(slice_node->attrs(), "begin_mask", &begin_mask)); in PartialStridedSliceShape()
599 if (!(begin_mask == 0 || begin_mask == 1) || in PartialStridedSliceShape()
608 if (begin_mask == 1) { in PartialStridedSliceShape()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DStridedSlice.pbtxt38 name: "begin_mask"
DResourceStridedSliceAssign.pbtxt38 name: "begin_mask"
DStridedSliceGrad.pbtxt42 name: "begin_mask"
DTensorStridedSliceUpdate.pbtxt42 name: "begin_mask"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DStridedSlice.pbtxt38 name: "begin_mask"
DResourceStridedSliceAssign.pbtxt38 name: "begin_mask"
DStridedSliceGrad.pbtxt42 name: "begin_mask"
DTensorStridedSliceUpdate.pbtxt42 name: "begin_mask"

1234