| /external/tensorflow/tensorflow/lite/kernels/ |
| D | split_v.cc | 33 size_splits = GetInput(context, node, 1); in OpContext()
38 const TfLiteTensor* size_splits; member
50 void GetSizeSplitsVector(const TfLiteTensor* size_splits, in GetSizeSplitsVector() argument
52 const auto num_elements = NumElements(size_splits); in GetSizeSplitsVector()
54 size_splits_vector->push_back(GetTensorData<T>(size_splits)[i]); in GetSizeSplitsVector()
60 const TfLiteTensor* size_splits, in ResizeOutputTensors() argument
68 if (size_splits->type == kTfLiteInt32) { in ResizeOutputTensors()
69 GetSizeSplitsVector<int32_t>(size_splits, &size_splits_vector); in ResizeOutputTensors()
70 } else if (size_splits->type == kTfLiteInt64) { in ResizeOutputTensors()
71 GetSizeSplitsVector<int64_t>(size_splits, &size_splits_vector); in ResizeOutputTensors()
[all …]
|
| D | split_v_test.cc | 31 SplitVOpModel(const TensorData& input, const TensorData& size_splits, in SplitVOpModel() argument
34 size_splits_ = AddInput(size_splits); in SplitVOpModel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | split_op_test.py | 91 size_splits = array_ops.placeholder(dtype=dtypes.int32, shape=[None])
97 sess.run(array_ops.split(value, size_splits), {size_splits: [2, 2, 6]})
102 size_splits = array_ops.constant([2, 2, 6], dtype=dtypes.int32)
107 array_ops.split(value, size_splits, num=4)
109 r = self.evaluate(array_ops.split(value, size_splits, num=3))
135 size_splits = np.random.randint(2, 8, num_split, dtype=np.int32)
136 shape[split_dim] = np.sum(size_splits)
139 result = self.evaluate(array_ops.split(inp, size_splits, split_dim))
143 slices[split_dim] = slice(offset, offset + size_splits[i])
144 offset += size_splits[i]
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op_test.cc | 57 static Graph* MakeGraph(int split_dim, const std::vector<int64>& size_splits, in MakeGraph() argument
64 Tensor size_splits_tensor = test::AsTensor<int64>(size_splits); in MakeGraph()
70 .Attr("num_split", static_cast<int64>(size_splits.size())) in MakeGraph()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SplitV.pbtxt | 10 name: "size_splits"
30 `size_splits[i]`.
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 241 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>().setConstant(2); in TEST()
300 auto dx = SplitVGrad(x, size_splits, 1, dy0, dy1); in TEST()
|
| D | array_ops.cc | 611 const Tensor* size_splits = c->input_tensor(1); in __anon7c94107b0b02() local
621 } else if (size_splits == nullptr && c->ValueKnown(split_dimension)) { in __anon7c94107b0b02()
631 } else if (size_splits == nullptr && !c->ValueKnown(split_dimension)) { in __anon7c94107b0b02()
645 if (size_splits->dtype() == DT_INT32) { in __anon7c94107b0b02()
646 data = AsInt64<int32>(size_splits, size_splits->shape().dim_size(0)); in __anon7c94107b0b02()
648 data = AsInt64<int64>(size_splits, size_splits->shape().dim_size(0)); in __anon7c94107b0b02()
|
| D | ops.pbtxt | 37929 name: "size_splits"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1520 size_splits = ops.convert_to_tensor(num_or_size_splits)
1526 if size_splits._rank() == 0:
1532 size_splits_shape = size_splits._shape_tuple()
1539 value=value, size_splits=size_splits, axis=axis, num_split=num, name=name)
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_compatibility.md | 938 1: 1-D tensor (size_splits)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 332 value=conv, size_splits=sizes, axis=dim, num_split=3)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3609 …argspec: "args=[\'value\', \'size_splits\', \'axis\', \'num_split\', \'name\'], varargs=None, keyw…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3609 …argspec: "args=[\'value\', \'size_splits\', \'axis\', \'num_split\', \'name\'], varargs=None, keyw…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2775 func SplitV(scope *Scope, value tf.Output, size_splits tf.Output, axis tf.Output, num_split int64) …
2783 value, size_splits, axis,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 35099 name: "size_splits"
|
| D | ops_history.v2.pbtxt | 78999 name: "size_splits"
|
| D | ops_history.v1.pbtxt | 79626 name: "size_splits"
|