Home
last modified time | relevance | path

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

123

/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.h61 p->begin_mask <<= pad_count; in StridedSlicePadIndices()
63 p->begin_mask |= (1 << pad_count) - 1; in StridedSlicePadIndices()
76 const auto begin_mask = params.begin_mask; in StartForAxis() local
83 if (begin_mask & 1 << axis) { in StartForAxis()
170 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
185 op_params.begin_mask = begin_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc36 const int32 begin_mask, end_mask; member
48 int32 begin_mask; member
80 dense->begin_mask = 0; in BuildDenseSpec()
108 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
130 if (sparse.begin_mask & (1 << i)) { in BuildDenseSpec()
131 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
271 {dense_spec.begin_mask & (1 << i), dense_spec.end_mask & (1 << i)}}; in ValidateStridedSliceOp()
292 (dense_spec.begin_mask & (1 << i)) && (dense_spec.end_mask & (1 << i)); in 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/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()
351 old_lhs->shape(), begin_mask, end_mask, ellipsis_mask, in Compute()
399 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc112 const int32 begin_mask = 0; in BM_ValidateStridedSliceOp() local
119 &begin, &end, strides, input_shape, begin_mask, end_mask, ellipsis_mask, in BM_ValidateStridedSliceOp()
/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/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()
Dresolve_constant_strided_slice.cc57 op.begin_mask, op.end_mask, op.shrink_axis_mask, op.start_indices, in StridedSlice()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc292 tf_strided_slice_op.begin_mask().getSExtValue()), in matchAndRewrite()
307 int begin_mask = tf_strided_slice_op.begin_mask().getSExtValue(); in matchAndRewrite() local
309 op, rewriter, tf_strided_slice_op.begin(), begin_pad_val, &begin_mask); in matchAndRewrite()
323 rewriter.getI32IntegerAttr(begin_mask), in matchAndRewrite()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice.cc77 op_params.begin_mask = op_context->params->begin_mask; in BuildStridedSliceParams()
Dstrided_slice_test.cc55 std::initializer_list<int> strides_shape, int begin_mask, in TestStrideSlide() argument
93 TfLiteStridedSliceParams builtin_data = {begin_mask, end_mask, ellipsis_mask, in TestStrideSlide()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc579 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
581 GetNodeAttr(slice_node->attrs(), "begin_mask", &begin_mask)); in PartialStridedSliceShape()
592 if (!(begin_mask == 0 || begin_mask == 1) || in PartialStridedSliceShape()
601 if (begin_mask == 1) { in PartialStridedSliceShape()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DStridedSlice.pbtxt38 name: "begin_mask"
DResourceStridedSliceAssign.pbtxt38 name: "begin_mask"
DTensorStridedSliceUpdate.pbtxt42 name: "begin_mask"
DStridedSliceGrad.pbtxt42 name: "begin_mask"
DStridedSliceAssign.pbtxt44 name: "begin_mask"
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc74 op_params.begin_mask = op_context->params->begin_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1109 int begin_mask;
1124 begin_mask = other.begin_mask;
1141 begin_mask |= 1 << i;
1154 begin_mask = toco::port::ReverseBits32(static_cast<uint32>(begin_mask)) >>
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dstrided_slice.py65 begin_mask=parameters["begin_mask"],
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc413 int64 begin_mask; in StridedSliceGradHelper() local
419 GetNodeAttr(op.node()->attrs(), "begin_mask", &begin_mask)); in StridedSliceGradHelper()
429 StridedSliceGrad::BeginMask(begin_mask) in StridedSliceGradHelper()
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py909 begin_mask, end_mask = 0, 0
918 begin_mask |= (1 << index)
973 begin_mask=begin_mask,
1041 begin_mask=0, argument
1141 begin_mask=begin_mask,
1166 begin_mask=begin_mask,
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h342 int begin_mask; member
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py933 begin_mask = tensor.op.get_attr("begin_mask")
934 if begin_mask == 1:
944 not shrink_axis_mask and (not begin_mask or
945 (begin_mask == 1)) and

123