Home
last modified time | relevance | path

Searched refs:size_splits (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dsplit_v.cc37 size_splits = GetInput(context, node, 1); in OpContext()
42 const TfLiteTensor* size_splits; member
56 void GetSizeSplitsVector(const TfLiteTensor* size_splits, in GetSizeSplitsVector() argument
58 const auto num_elements = NumElements(size_splits); in GetSizeSplitsVector()
60 size_splits_vector->push_back(GetTensorData<T>(size_splits)[i]); in GetSizeSplitsVector()
66 const TfLiteTensor* size_splits, in ResizeOutputTensors() argument
74 if (size_splits->type == kTfLiteInt32) { in ResizeOutputTensors()
75 GetSizeSplitsVector<int32_t>(size_splits, &size_splits_vector); in ResizeOutputTensors()
76 } else if (size_splits->type == kTfLiteInt64) { in ResizeOutputTensors()
77 GetSizeSplitsVector<int64_t>(size_splits, &size_splits_vector); in ResizeOutputTensors()
[all …]
Dsplit_v_test.cc41 SplitVOpModel(const TensorData& input, const TensorData& size_splits, in SplitVOpModel() argument
46 size_splits_ = AddInput(size_splits); in SplitVOpModel()
48 size_splits_ = AddConstInput(size_splits, size_splits_data); in SplitVOpModel()
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dsplit_op_test.py87 size_splits = array_ops.placeholder(dtype=dtypes.int32, shape=[None])
93 sess.run(array_ops.split(value, size_splits), {size_splits: [2, 2, 6]})
99 size_splits = array_ops.constant([2, 2, 6], dtype=dtypes.int32)
104 array_ops.split(value, size_splits, num=4)
106 r = self.evaluate(array_ops.split(value, size_splits, num=3))
132 size_splits = np.random.randint(2, 8, num_split, dtype=np.int32)
133 shape[split_dim] = np.sum(size_splits)
136 result = self.evaluate(array_ops.split(inp, size_splits, split_dim))
140 slices[split_dim] = slice(offset, offset + size_splits[i])
141 offset += size_splits[i]
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/
Dops_defs.py61 size_splits=[left_padding_size, -1],
66 size_splits=[-1, right_padding_size],
72 size_splits=[-1, left_padding_size],
77 size_splits=[right_padding_size, -1],
108 size_splits=[left_padding_size, -1, right_padding_size],
120 size_splits=[-1, left_padding_size],
125 size_splits=[-1, right_padding_size],
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SplitV.pbtxt10 name: "size_splits"
30 `size_splits[i]`.
/external/tensorflow/tensorflow/core/kernels/
Dsplit_v_op_test.cc60 static Graph* MakeGraph(int split_dim, const std::vector<int64_t>& size_splits, in MakeGraph() argument
67 Tensor size_splits_tensor = test::AsTensor<int64_t>(size_splits); in MakeGraph()
73 .Attr("num_split", static_cast<int64_t>(size_splits.size())) in MakeGraph()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-173.md14 size_splits=[-1, -2]
19 This occurs whenever `size_splits` contains more than one value and at least one value is negative.
/external/ComputeLibrary/src/graph/nodes/
DSplitLayerNode.cpp35 SplitLayerNode::SplitLayerNode(unsigned int num_splits, int axis, std::vector<int> size_splits) in SplitLayerNode() argument
36 : _num_splits(num_splits), _axis(axis), _size_splits(size_splits) in SplitLayerNode()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSplitV.pbtxt8 name: "size_splits"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSplitV.pbtxt8 name: "size_splits"
/external/ComputeLibrary/arm_compute/graph/nodes/
DSplitLayerNode.h47 …SplitLayerNode(unsigned int num_splits, int axis = 0, std::vector<int> size_splits = std::vector<i…
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dtfr_gen_test.py139 value=lhs, size_splits=[rhs, -1], axis=0, num_split=2)
141 value=lhs, size_splits=[rhs, rhs], axis=1, num_split=2)
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc241 std::vector<Tensor> SplitVGrad(const Tensor& x, const Tensor& size_splits, in SplitVGrad() argument
264 {"size_splits:0", size_splits}, in SplitVGrad()
297 Tensor size_splits(DT_INT64, {2}); in TEST() local
298 size_splits.flat<int64_t>().setConstant(2); in TEST()
300 auto dx = SplitVGrad(x, size_splits, 1, dy0, dy1); in TEST()
Darray_ops.cc618 const Tensor* size_splits = c->input_tensor(1); in __anon091a71b70b02() local
628 } else if (size_splits == nullptr && c->ValueKnown(split_dimension)) { in __anon091a71b70b02()
638 } else if (size_splits == nullptr && !c->ValueKnown(split_dimension)) { in __anon091a71b70b02()
652 if (size_splits->dtype() == DT_INT32) { in __anon091a71b70b02()
653 data = AsInt64<int32>(size_splits, size_splits->shape().dim_size(0)); in __anon091a71b70b02()
656 AsInt64<int64_t>(size_splits, size_splits->shape().dim_size(0)); in __anon091a71b70b02()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dops.mlir2205 %size_splits = arith.constant dense<[]> : tensor<0xi32>
2208 …"tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 0 : i32} : (tensor<16xf32>, tensor<0x…
2215 %size_splits = arith.constant dense<[4, 4, 4, 4]> : tensor<4xi32>
2218 …%0, %1 = "tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 4 : i32} : (tensor<16xf32>, …
2225 %size_splits = arith.constant dense<[[8, 8], [2, 2]]> : tensor<2x2xi32>
2228 …%0 = "tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 1 : i32} : (tensor<16x4x4xf32>, …
2234 func.func @testSplitVOpWithBadSizeSplitsUnrankedTensorType(%arg0: tensor<16x4x4xf32>, %size_splits:…
2237 …%0 = "tfl.split_v"(%arg0, %size_splits, %split_dim) {num_splits = 1 : i32} : (tensor<16x4x4xf32>, …
2244 %size_splits = arith.constant dense<[-2]> : tensor<1xi32>
2246 …// expected-error @+1 {{'tfl.split_v' op elements of 'size_splits' should be greater than or equal…
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2423 if (!matchPattern(op.size_splits(), m_Constant(&size_splits_attr))) in verify()
2427 auto size_splits_type = op.size_splits().getType().cast<RankedTensorType>(); in verify()
2437 llvm::SmallVector<int64_t, 4> size_splits; in verify() local
2438 size_splits.reserve(num_splits); in verify()
2446 size_splits.push_back(size_split); in verify()
2464 size_splits[negative_size_split_loc] = dim_size - total_size_splits; in verify()
2475 &size_splits](int64_t i) { in verify()
2477 size_splits[i]); in verify()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc619 const auto* size_splits = size_splits_tensor.data.i32; in ComputeSplitVUnknownSplitSize() local
624 if (size_splits[i] == -1) { in ComputeSplitVUnknownSplitSize()
627 sum_of_known_split_sizes += size_splits[i]; in ComputeSplitVUnknownSplitSize()
1333 const auto* size_splits = size_splits_tensor.data.i32; in TransformSplitVIntoSupportedOps() local
1344 int split_size = size_splits[split_index] == -1 in TransformSplitVIntoSupportedOps()
1346 : size_splits[split_index]; in TransformSplitVIntoSupportedOps()
3286 const TfLiteTensor& size_splits = context->tensors[node->inputs->data[1]]; in Validate() local
3291 size_splits.type == kTfLiteInt32 && size_splits.dims->size == 1 && in Validate()
3292 size_splits.allocation_type == kTfLiteMmapRo; in Validate()
3303 Expect(std::all_of(size_splits.data.i32, in Validate()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py2186 size_splits = ops.convert_to_tensor(num_or_size_splits)
2188 if size_splits._rank() == 0:
2194 size_splits_shape = size_splits._shape_tuple()
2202 value=value, size_splits=size_splits, axis=axis, num_split=num, name=name)
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc320 auto splits = splitv_op.size_splits().getDefiningOp(); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc311 tf_splitv_op.size_splits(), tf_splitv_op.split_dim(), num_split); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc4383 std::vector<int32> size_splits(n); in TEST_F() local
4385 size_splits.push_back(dims[dim] / n); in TEST_F()
4387 size_splits.push_back(dims[dim] - (n - 1) * (dims[dim] / n)); in TEST_F()
4391 .Input(test::AsTensor<int32>(size_splits)) in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md3001 %output = tf.SplitV(%value, %size_splits, %split_dim) {num_split}
3007 %output = lower_splitv_op(%value, %size_splits.as_constant(), %split_dim.as_constant())
4931 %output = tfl.split_v(%value, %size_splits, %split_dim) {num_splits}
4937 %output = lower_splitv_op(%value, %size_splits.as_constant(), %split_dim.as_constant())
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1571 op.size_splits().getType().dyn_cast<RankedTensorType>(); in verify()
1591 if (!matchPattern(op.size_splits(), m_Constant(&split_sizes_attr))) in verify()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc3459 auto size_splits = in TEST_F() local
3463 conv2d, size_splits, axis, 3); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1574 if (!matchPattern(tf_splitv_op.size_splits(), in matchAndRewrite()

12