Home
last modified time | relevance | path

Searched refs:end_mask (Results 1 – 25 of 94) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt1763 …,begin_mask=None,constant_indices=False,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1764 …,begin_mask=None,constant_indices=False,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1765 …0],begin_mask=None,constant_indices=False,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
1766 …0],begin_mask=None,constant_indices=False,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
1767 …0],begin_mask=None,constant_indices=False,dtype=tf.int64,end=[12,2,2,5],end_mask=None,index_type=t…
1768 …0],begin_mask=None,constant_indices=False,dtype=tf.int64,end=[12,2,2,5],end_mask=None,index_type=t…
1769 …],begin_mask=None,constant_indices=True,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1770 …],begin_mask=None,constant_indices=True,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1771 …,0],begin_mask=None,constant_indices=True,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
1772 …,0],begin_mask=None,constant_indices=True,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h63 p->end_mask <<= pad_count; in StridedSlicePadIndices()
65 p->end_mask |= (1 << pad_count) - 1; in StridedSlicePadIndices()
125 const auto end_mask = params.end_mask; in StopForAxis() local
147 if (end_mask & (1 << axis)) { in StopForAxis()
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
200 op_params.end_mask = end_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dstrided_slice_builder.cc52 int end_mask = params->end_mask; in PopulateSubGraph() local
56 end_mask = end_mask << (4 - original_mask_size); in PopulateSubGraph()
62 kScalarShape, reinterpret_cast<char*>(&end_mask), sizeof(end_mask)); in PopulateSubGraph()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dfuse-tftext.mlir29 …, %5, %16) {begin_mask = 1 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
31 … %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 1 : i64, new_ax…
34 … %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
41 … %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
46 … %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 1 : i64, new_ax…
47 …, %5, %16) {begin_mask = 1 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
54 … %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
56 … %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
63 … %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
65 … %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_ax…
[all …]
Dprepare-tf.mlir459 …g0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, end_mask = 0 : i64, new_ax…
464 …, %[[CST]], %[[CST_0]]) {begin_mask = 3 : i64, ellipsis_mask = 0 : i64, end_mask = 3 : i64, new_ax…
472 …, %cst, %cst_0, %cst_1) {begin_mask = 6 : i64, ellipsis_mask = 1 : i64, end_mask = 4 : i64, new_ax…
478 …%[[CST_0]], %[[CST_1]]) {begin_mask = 7 : i64, ellipsis_mask = 0 : i64, end_mask = 5 : i64, new_ax…
485 …g0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 2 : i64, end_mask = 0 : i64, new_ax…
490 …, %[[CST]], %[[CST_0]]) {begin_mask = 6 : i64, ellipsis_mask = 0 : i64, end_mask = 6 : i64, new_ax…
497 …g0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, end_mask = 0 : i64, new_ax…
502 …, %[[CST]], %[[CST_0]]) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, end_mask = 0 : i64, new_ax…
509 …0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask = 0 : i64, end_mask = 15 : i64, new_a…
514 …0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask = 0 : i64, end_mask = 15 : i64, new_a…
[all …]
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc38 const int32 begin_mask, end_mask; member
51 int32 end_mask; member
90 dense->end_mask = 0; in BuildDenseSpec()
118 dense->end_mask |= (1 << full_index); in BuildDenseSpec()
145 if (sparse.end_mask & (1 << i)) { in BuildDenseSpec()
146 dense->end_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()
390 shape_spec->end_dense_mask = dense_spec.end_mask; in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc86 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceOp()
105 context->input(0).shape(), begin_mask, end_mask, in Compute()
184 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
194 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_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
288 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_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.cc110 const int32 end_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.cc568 int32 end_mask, int32 ellipsis_mask, in StridedSliceGrad() argument
585 {"end_mask", end_mask}, in StridedSliceGrad()
610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() argument
628 {"end_mask", end_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/lite/kernels/
Dstrided_slice.cc77 op_params.end_mask = op_context->params->end_mask; in BuildStridedSliceParams()
94 op_params.end_mask |= (1 << i); in BuildStridedSliceParams()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_strided_slice_attributes.cc113 op->end_mask = in Run()
114 PadAttributeArray(&stop_array, input_array.shape().dims(), op->end_mask); in Run()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt17 `end[i]` is like `begin` with the exception that `end_mask` is
40 name: "end_mask"
104 `-1` if `stride < 0`. `end_mask` is analogous but produces the number
127 end_mask = 1<<5 = 32
161 `end_mask` are also set.
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc573 uint64_t revised_end_mask = strided_slice_op.end_mask(); in RewriteNewAxisMask()
646 int64_t end_mask = strided_slice_op.end_mask(); in RewriteEllipsisMask() local
663 if ((end_mask >> index) & 1) revised_end_mask |= (1 << new_index); in RewriteEllipsisMask()
691 if ((end_mask >> index) & 1) revised_end_mask |= (1 << new_index); in RewriteEllipsisMask()
800 int end_mask = strided_slice_op.end_mask(); in matchAndRewrite() local
805 &end_mask); in matchAndRewrite()
812 end_mask == strided_slice_op.end_mask()) { in matchAndRewrite()
836 rewriter.getIntegerAttr(attribute_type, end_mask), in matchAndRewrite()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dstrided_slice_test.cc31 int end_mask, int shrink_axis_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
176 builder->getI64IntegerAttr(end_mask), in CreateStridedSliceOp()
780 int64_t end_mask; in ConvertKerasLSTMLayer() local
784 end_mask = 6; in ConvertKerasLSTMLayer()
789 end_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.cc76 op_params.end_mask = op_context->params->end_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
589 TF_RETURN_IF_ERROR(GetNodeAttr(slice_node->attrs(), "end_mask", &end_mask)); in PartialStridedSliceShape()
600 !(end_mask == 0 || end_mask == 1) || ellipsis_mask != 0 || in PartialStridedSliceShape()
620 if (end_mask == 1) { in PartialStridedSliceShape()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DStridedSlice.pbtxt45 name: "end_mask"
DResourceStridedSliceAssign.pbtxt45 name: "end_mask"
DStridedSliceGrad.pbtxt49 name: "end_mask"
DTensorStridedSliceUpdate.pbtxt49 name: "end_mask"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DStridedSlice.pbtxt45 name: "end_mask"
DResourceStridedSliceAssign.pbtxt45 name: "end_mask"
DStridedSliceGrad.pbtxt49 name: "end_mask"
DTensorStridedSliceUpdate.pbtxt49 name: "end_mask"

1234