Home
last modified time | relevance | path

Searched refs:begin_mask (Results 1 – 25 of 53) 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.cc83 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceOp()
103 context->input(0).shape(), begin_mask, end_mask, in Compute()
181 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
190 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceGradOp()
229 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
274 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp
283 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceAssignOp()
325 old_lhs->shape(), begin_mask, end_mask, ellipsis_mask, in Compute()
372 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/kernels/
Dstrided_slice.cc84 return op_context->params->begin_mask & (1 << idx) in GetBeginValueAtIndex()
202 int begin_mask = op_context.params->begin_mask << (4 - op_context.dims); in Eval() local
208 begin_mask, end_mask, shrink_axis_mask, starts, stops, strides); in Eval()
Dstrided_slice_test.cc33 std::initializer_list<int> strides_shape, int begin_mask, in StridedSliceOpModel() argument
43 CreateStridedSliceOptions(builder_, begin_mask, end_mask, ellipsis_mask, in StridedSliceOpModel()
/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/core/common_runtime/
Dshape_refiner.cc561 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
563 GetNodeAttr(slice_node->attrs(), "begin_mask", &begin_mask)); in PartialStridedSliceShape()
574 if (!(begin_mask == 0 || begin_mask == 1) || in PartialStridedSliceShape()
583 if (begin_mask == 1) { in PartialStridedSliceShape()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1062 int begin_mask;
1077 begin_mask = other.begin_mask;
1094 begin_mask |= 1 << i;
1107 begin_mask = toco::port::ReverseBits32(static_cast<uint32>(begin_mask)) >>
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc2584 key: "begin_mask" in CreateImagenetDatasetFunctions()
2709 key: "begin_mask" in CreateImagenetDatasetFunctions()
2891 key: "begin_mask" in CreateImagenetDatasetFunctions()
3036 key: "begin_mask" in CreateImagenetDatasetFunctions()
3270 key: "begin_mask" in CreateImagenetDatasetFunctions()
3415 key: "begin_mask" in CreateImagenetDatasetFunctions()
3708 key: "begin_mask" in CreateImagenetDatasetFunctions()
3850 key: "begin_mask" in CreateImagenetDatasetFunctions()
3992 key: "begin_mask" in CreateImagenetDatasetFunctions()
4529 key: "begin_mask" in CreateImagenetDatasetFunctions()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py598 begin_mask, end_mask = 0, 0
610 begin_mask |= (1 << index)
663 begin_mask=begin_mask,
731 begin_mask=0, argument
831 begin_mask=begin_mask,
855 begin_mask=begin_mask,
Dresource_variable_ops.py1343 def _strided_slice_assign(self, begin, end, strides, value, name, begin_mask, argument
1355 begin_mask=begin_mask,
Dvariables.py2119 begin_mask, argument
2130 begin_mask=begin_mask,
Darray_grad.py273 begin_mask=op.get_attr("begin_mask"),
292 begin_mask=op.get_attr("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/lite/c/
Dbuiltin_op_data.h307 int begin_mask; member
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py907 begin_mask = tensor.op.get_attr("begin_mask")
908 if begin_mask == 1:
918 not shrink_axis_mask and (not begin_mask or
919 (begin_mask == 1)) and
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1849 begin_mask = pfor_input.get_attr("begin_mask")
1858 begin_mask = begin_mask << 1 | 1
1869 begin_mask=begin_mask,
1883 begin_mask = pfor_input.get_attr("begin_mask")
1893 begin_mask = begin_mask << 1 | 1
1905 begin_mask=begin_mask,
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc541 params->begin_mask = schema_params->begin_mask(); in ParseOpData()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc457 op.begin_mask = 1; in TEST_F()
468 EXPECT_EQ(op.begin_mask, output_toco_op->begin_mask); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1527 int begin_mask = 0; in MaybeUpdateNodeContextOutput() local
1529 begin_mask = node.attr().at("begin_mask").i(); in MaybeUpdateNodeContextOutput()
1535 if (begin_mask < 0 || begin_mask > 1 || end_mask < 0 || end_mask > 1) { in MaybeUpdateNodeContextOutput()
1540 if (begin_mask == 0) { in MaybeUpdateNodeContextOutput()

123