Home
last modified time | relevance | path

Searched refs:shrink_axis_mask (Results 1 – 25 of 87) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt1763 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1764 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1765 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1766 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1767 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1768 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1769 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1770 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1771 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
1772 …[12,2,2,5],end_mask=None,index_type=tf.int32,input_shape=[12,2,2,5],shrink_axis_mask=None,strides=…
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dstrided_slice_builder.cc53 int shrink_axis_mask = params->shrink_axis_mask; in PopulateSubGraph() local
57 shrink_axis_mask = shrink_axis_mask << (4 - original_mask_size); in PopulateSubGraph()
65 kScalarShape, reinterpret_cast<char*>(&shrink_axis_mask), in PopulateSubGraph()
66 sizeof(shrink_axis_mask)); in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h59 p->shrink_axis_mask <<= pad_count; in StridedSlicePadIndices()
126 const auto shrink_axis_mask = params.shrink_axis_mask; in StopForAxis() local
135 const bool shrink_axis = shrink_axis_mask & (1 << axis); in StopForAxis()
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
202 op_params.shrink_axis_mask = shrink_axis_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc40 const int32 new_axis_mask, shrink_axis_mask; member
74 int32 shrink_axis_mask; member
91 dense->shrink_axis_mask = 0; in BuildDenseSpec()
151 if (sparse.shrink_axis_mask & (1 << i)) { in BuildDenseSpec()
154 dense->shrink_axis_mask |= (1 << full_index); in BuildDenseSpec()
173 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp() argument
225 shrink_axis_mask}; in ValidateStridedSliceOp()
281 bool shrink_i = (dense_spec.shrink_axis_mask & (1 << i)); in ValidateStridedSliceOp()
391 shape_spec->shrink_axis_dense_mask = dense_spec.shrink_axis_mask; in ValidateStridedSliceOp()
421 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp() argument
[all …]
Dstrided_slice_op.h68 int32 new_axis_mask, int32 shrink_axis_mask,
80 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dfuse-tftext.mlir29 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
31 …lipsis_mask = 0 : i64, end_mask = 1 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
34 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64} : …
41 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64} : …
46 …lipsis_mask = 0 : i64, end_mask = 1 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
47 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
54 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64} : …
56 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64} : …
63 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64} : …
65 …lipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64} : …
[all …]
Dprepare-tf.mlir459 …lipsis_mask = 1 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
464 …lipsis_mask = 0 : i64, end_mask = 3 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
472 …lipsis_mask = 1 : i64, end_mask = 4 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
478 …lipsis_mask = 0 : i64, end_mask = 5 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
485 …lipsis_mask = 2 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
490 …lipsis_mask = 0 : i64, end_mask = 6 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
497 …lipsis_mask = 1 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
502 …lipsis_mask = 1 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
509 …ipsis_mask = 0 : i64, end_mask = 15 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
514 …ipsis_mask = 0 : i64, end_mask = 15 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : …
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc90 context->GetAttr("shrink_axis_mask", &shrink_axis_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
198 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceGradOp()
233 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
279 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
292 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceAssignOp()
343 new_axis_mask, shrink_axis_mask, &processing_shape, in Compute()
391 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc113 const int32 shrink_axis_mask = 0; in BM_ValidateStridedSliceOp() local
117 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc570 int32 shrink_axis_mask) { in StridedSliceGrad() argument
587 {"shrink_axis_mask", shrink_axis_mask}, in StridedSliceGrad()
611 int32 shrink_axis_mask) { in StridedSliceGradGrad() argument
630 {"shrink_axis_mask", shrink_axis_mask}, in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
668 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
680 new_axis_mask, shrink_axis_mask); in TEST()
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
695 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
707 new_axis_mask, shrink_axis_mask); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice.cc78 op_params.shrink_axis_mask = op_context->params->shrink_axis_mask; in BuildStridedSliceParams()
103 const bool shrink_axis = op_context->params->shrink_axis_mask & (1 << idx); in CheckOutputSize()
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc79 op_params.shrink_axis_mask = op_context->params->shrink_axis_mask; in BuildStridedSliceParams()
121 const bool shrink_axis = op_context->params->shrink_axis_mask & (1 << idx); in ResizeOutputTensor()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt66 name: "shrink_axis_mask"
72 `shrink_axis_mask` being 2.
115 `shrink_axis_mask`.
130 shrink_axis_mask = 1<<0 = 1
139 also set the appropriate bit in `shrink_axis_mask`.
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dstrided_slice_test.cc31 int end_mask, int shrink_axis_mask) { in StridedSliceOpModel() argument
41 /*new_axis_mask*/ 0, shrink_axis_mask) in StridedSliceOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc166 int64_t shrink_axis_mask, OpBuilder* builder) { in CreateStridedSliceOp() argument
179 builder->getI64IntegerAttr(shrink_axis_mask)); in CreateStridedSliceOp()
781 int64_t shrink_axis_mask; in ConvertKerasLSTMLayer() local
785 shrink_axis_mask = 1; in ConvertKerasLSTMLayer()
790 shrink_axis_mask = 2; in ConvertKerasLSTMLayer()
797 shrink_axis_mask, builder); in ConvertKerasLSTMLayer()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DStridedSlice.pbtxt66 name: "shrink_axis_mask"
DResourceStridedSliceAssign.pbtxt66 name: "shrink_axis_mask"
DStridedSliceGrad.pbtxt70 name: "shrink_axis_mask"
DTensorStridedSliceUpdate.pbtxt70 name: "shrink_axis_mask"
DStridedSliceAssign.pbtxt72 name: "shrink_axis_mask"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DStridedSlice.pbtxt66 name: "shrink_axis_mask"
DResourceStridedSliceAssign.pbtxt66 name: "shrink_axis_mask"
DStridedSliceGrad.pbtxt70 name: "shrink_axis_mask"
DTensorStridedSliceUpdate.pbtxt70 name: "shrink_axis_mask"
DStridedSliceAssign.pbtxt72 name: "shrink_axis_mask"

1234