Home
last modified time | relevance | path

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

12

/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/kernels/internal/
Dstrided_slice_logic.h58 p->shrink_axis_mask <<= pad_count; in StridedSlicePadIndices()
116 const auto shrink_axis_mask = params.shrink_axis_mask; in StopForAxis() local
121 const bool shrink_axis = shrink_axis_mask & (1 << axis); in StopForAxis()
170 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
189 op_params.shrink_axis_mask = shrink_axis_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc38 const int32 new_axis_mask, shrink_axis_mask; member
66 int32 shrink_axis_mask; member
82 dense->shrink_axis_mask = 0; in BuildDenseSpec()
139 if (sparse.shrink_axis_mask & (1 << i)) { in BuildDenseSpec()
141 dense->shrink_axis_mask |= (1 << full_index); in BuildDenseSpec()
156 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp() argument
208 shrink_axis_mask}; in ValidateStridedSliceOp()
264 bool shrink_i = (dense_spec.shrink_axis_mask & (1 << i)); in ValidateStridedSliceOp()
379 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp() argument
387 end_mask_spec, ellipsis_mask, new_axis_mask, shrink_axis_mask, in ValidateStridedSliceOp()
Dstrided_slice_op.h47 int32 new_axis_mask, int32 shrink_axis_mask,
58 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
/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/core/kernels/
Dstrided_slice_op.cc88 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceOp()
104 ellipsis_mask, new_axis_mask, shrink_axis_mask, in Compute()
182 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
195 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceGradOp()
230 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
275 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
288 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceAssignOp()
326 new_axis_mask, shrink_axis_mask, &processing_shape, in Compute()
373 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc116 const int32 shrink_axis_mask = 0; in BM_ValidateStridedSliceOp() local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc117 const bool shrink_axis = op_context->params->shrink_axis_mask & (1 << idx); in ResizeOutputTensor()
204 int shrink_axis_mask = op_context.params->shrink_axis_mask in Eval() local
208 begin_mask, end_mask, shrink_axis_mask, starts, stops, strides); in Eval()
Dstrided_slice_test.cc35 int shrink_axis_mask) { in StridedSliceOpModel() argument
44 new_axis_mask, shrink_axis_mask) in StridedSliceOpModel()
/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
139 also set the appropriate bit in `shrink_axis_mask`.
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc2608 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
2733 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
2915 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
3060 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
3294 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
3439 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
3732 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
3874 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
4016 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
4553 key: "shrink_axis_mask" in CreateImagenetDatasetFunctions()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py597 new_axis_mask, shrink_axis_mask = 0, 0
637 shrink_axis_mask |= (1 << index)
665 shrink_axis_mask=shrink_axis_mask,
735 shrink_axis_mask=0, argument
835 shrink_axis_mask=shrink_axis_mask)
859 shrink_axis_mask=shrink_axis_mask)
Dresource_variable_ops.py1345 shrink_axis_mask): argument
1359 shrink_axis_mask=shrink_axis_mask))
Dvariables.py2123 shrink_axis_mask): argument
2134 shrink_axis_mask=shrink_axis_mask)
Darray_grad.py277 shrink_axis_mask=op.get_attr("shrink_axis_mask")), None, None, None
296 shrink_axis_mask=op.get_attr("shrink_axis_mask"))
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1066 int shrink_axis_mask;
1081 shrink_axis_mask = other.shrink_axis_mask;
1117 shrink_axis_mask =
1118 toco::port::ReverseBits32(static_cast<uint32>(shrink_axis_mask)) >>
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc417 int64 shrink_axis_mask; in StridedSliceGradHelper() local
426 GetNodeAttr(op.node()->attrs(), "shrink_axis_mask", &shrink_axis_mask)); in StridedSliceGradHelper()
433 .ShrinkAxisMask(shrink_axis_mask))); in StridedSliceGradHelper()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h311 int shrink_axis_mask; member
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc561 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
570 GetNodeAttr(slice_node->attrs(), "shrink_axis_mask", &shrink_axis_mask)); in PartialStridedSliceShape()
576 new_axis_mask != 0 || shrink_axis_mask != 0) { in PartialStridedSliceShape()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_strided_slice.cc57 op.begin_mask, op.end_mask, op.shrink_axis_mask, op.start_indices, in StridedSlice()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc545 params->shrink_axis_mask = schema_params->shrink_axis_mask(); in ParseOpData()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1853 shrink_axis_mask = pfor_input.get_attr("shrink_axis_mask")
1862 shrink_axis_mask <<= 1
1873 shrink_axis_mask=shrink_axis_mask), True)
1887 shrink_axis_mask = pfor_input.get_attr("shrink_axis_mask")
1897 shrink_axis_mask <<= 1
1909 shrink_axis_mask=shrink_axis_mask), True)
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc461 op.shrink_axis_mask = 2; in TEST_F()
473 EXPECT_EQ(op.shrink_axis_mask, output_toco_op->shrink_axis_mask); in TEST_F()
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py916 shrink_axis_mask = tensor.op.get_attr("shrink_axis_mask")
918 not shrink_axis_mask and (not begin_mask or
/external/tensorflow/tensorflow/compiler/tests/
Dnary_ops_test.py242 return array_ops.strided_slice_grad(*x, shrink_axis_mask=0x4,

12