| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | split_op.cc | 57 int32 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile() local
59 OP_REQUIRES(ctx, 0 <= split_dim && split_dim < input_shape.dims(), in Compile()
71 ctx, input_shape.dim_size(split_dim) % num_split == 0, in Compile()
75 split_dim_orig, " (size = ", input_shape.dim_size(split_dim), ") ", in Compile()
80 const int32 slice_size = input_shape.dim_size(split_dim) / num_split; in Compile()
99 begin[split_dim] = i * slice_size; in Compile()
100 limits[split_dim] = (i + 1) * slice_size; in Compile()
118 int32 split_dim; in Compile() local
122 split_dim = literal_index.Get<int>({}); in Compile()
131 ctx, 0 <= split_dim && split_dim < input_shape.dims(), in Compile()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_op.cc | 53 const int32 split_dim = in ComputeEasyCases() local
58 context, 0 <= split_dim && split_dim < input_shape.dims(), in ComputeEasyCases()
68 OP_REQUIRES(context, input_shape.dim_size(split_dim) % num_split == 0, in ComputeEasyCases()
72 split_dim, " (size = ", input_shape.dim_size(split_dim), in ComputeEasyCases()
90 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) { in ComputeEasyCases()
103 const TensorShape& input_shape, int32 split_dim) const { in SetDims()
107 for (int i = 0; i < split_dim; ++i) { in SetDims()
114 static_cast<IndexType>(input_shape.dim_size(split_dim)); in SetDims()
117 for (int i = split_dim + 1; i < input_shape.dims(); ++i) { in SetDims()
140 const int32 split_dim = in Compute() local
[all …]
|
| D | split_op_test.cc | 29 static Graph* MakeGraph(int split_dim, int num_split, in MakeGraph() argument
33 in_shape.set_dim(split_dim, in_shape.dim_size(split_dim) * num_split); in MakeGraph()
36 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); in MakeGraph()
61 #define BM_SPLIT_2D(split_dim, num_split, chunk_size0, chunk_size1) \ argument
63 BM_Split_2d_##split_dim##_##num_split##_##chunk_size0##_##chunk_size1( \
70 num_split, split_dim, chunk_size0, chunk_size1); \
73 auto g = MakeGraph(split_dim, num_split, {chunk_size0, chunk_size1}); \
78 BM_Split_2d_##split_dim##_##num_split##_##chunk_size0##_##chunk_size1);
|
| D | split_v_op.cc | 60 const int32 split_dim = in ComputeEasyCases() local
84 context, 0 <= split_dim && split_dim < input.dims(), in ComputeEasyCases()
89 Tlen input_size_split_dim = input_shape.dim_size(split_dim); in ComputeEasyCases()
143 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) { in ComputeEasyCases()
157 const TensorShape& input_shape, const int32 split_dim) const { in SetDims()
161 for (int i = 0; i < split_dim; ++i) { in SetDims()
168 static_cast<IndexType>(input_shape.dim_size(split_dim)); in SetDims()
171 for (int i = split_dim + 1; i < input_shape.dims(); ++i) { in SetDims()
195 const int32 split_dim = in Compute() local
211 Base::template SetDims<Eigen::DenseIndex>(input_shape, split_dim); in Compute()
[all …]
|
| D | split_v_op_test.cc | 57 static Graph* MakeGraph(int split_dim, const std::vector<int64>& size_splits, in MakeGraph() argument
63 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); in MakeGraph()
91 #define BM_SPLITV_2D(split_dim, num_split, total_size0, total_size1) \ argument
93 BM_SplitV_2d_##split_dim##_##num_split##_##total_size0##_##total_size1( \
101 num_split, split_dim, total_size0, total_size1); \
105 split_dim, \
106 GenerateRandomIntsWithSum(total_size_vec[split_dim], num_split), \
112 BM_SplitV_2d_##split_dim##_##num_split##_##total_size0##_##total_size1);
|
| D | sparse_split_op.cc | 33 const int64 split_dim = context->input(0).scalar<int64>()(); in Compute() local
53 input_shape.dim_size(0) && split_dim < input_shape.vec<int64>().size(), in Compute()
56 input_shape.vec<int64>().size(), "), got ", split_dim)); in Compute()
60 num_split_ >= 1 && num_split_ <= input_shape.vec<int64>()(split_dim), in Compute()
63 input_shape.vec<int64>()(split_dim), "), got ", in Compute()
69 sparse::SparseTensor::Split<T>(sparse_tensor, split_dim, num_split_); in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | split_op_test.py | 128 split_dim = np.random.randint(-5, 5)
134 shape[split_dim] = np.sum(size_splits)
137 result = self.evaluate(array_ops.split(inp, size_splits, split_dim))
141 slices[split_dim] = slice(offset, offset + size_splits[i])
160 split_dim = 1
163 result = self.evaluate(array_ops.split(inp, size_splits, split_dim))
167 slices[split_dim] = slice(offset, offset + size_splits[i])
265 split_dim = np.random.randint(-5, 5)
270 shape[split_dim] = np.random.randint(2, 5) * num_split
275 value=inp, num_or_size_splits=num_split, axis=split_dim))
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Split.pbtxt | 4 name: "split_dim"
21 except along `split_dim`, where their sizes are
22 `values.shape[split_dim] / num_split`.
29 `value.shape[split_dim]`.
|
| D | api_def_SparseSplit.pbtxt | 4 name: "split_dim"
52 If the `shape[split_dim]` is not an integer multiple of `num_split`. Slices
53 `[0 : shape[split_dim] % num_split]` gets one extra dimension.
54 For example, if `split_dim = 1` and `num_split = 2` and the input is
|
| D | api_def_SplitV.pbtxt | 13 dimension. Must sum to the dimension of value along split_dim.
18 name: "split_dim"
29 except along `split_dim`, where their sizes are
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 170 const int split_dim,
496 const int split_dim, in Split() argument
514 const int split_dim_size = input_tensor.shape()[split_dim]; in Split()
520 CHECK(split_dim >= 0 && split_dim < num_dim) << "num_dim must be in " in Split()
526 const int dim = input_tensor.indices().matrix<int64>()(i, split_dim); in Split()
541 output_shapes[i].set_dim(split_dim, size); in Split()
546 const int dim = input_indices_t(i, split_dim); in Split()
553 (j == split_dim) in Split()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 81 split_dim = constant_op.constant(1)
83 x1, x2, x3 = array_ops.split(value, 3, axis=split_dim)
104 split_dim = constant_op.constant(1)
106 result = array_ops.split(value, 1, axis=split_dim)
128 split_dim = constant_op.constant(1, dtype=dtypes.int64)
134 split_dim, indices, values, shape, num_split=2)
|
| D | core_test.py | 449 split_dim = 1
454 inputs=[constant_op.constant(split_dim),
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker_test.cc | 129 auto split_dim = Const(scope, 1, {}); in TEST() local
130 auto y = Split(scope, split_dim, x, /* num_split */ 2); in TEST()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1107 Tensor split_dim = test::AsTensor<int32>({1, 2}); in TEST() local
1108 op.input_tensors[0] = &split_dim; in TEST()
1110 split_dim = test::AsScalar<int32>(1); in TEST()
1119 split_dim = test::AsScalar<int32>(3); in TEST()
1125 split_dim = test::AsScalar<int32>(-1); in TEST()
1131 split_dim = test::AsScalar<int32>(-2); in TEST()
1133 split_dim = test::AsScalar<int32>(-4); in TEST()
|
| D | array_ops.cc | 422 int64 split_dim = c->Value(split_dimension); in __anondb9326b20802() local
423 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anondb9326b20802()
426 c->Divide(c->Dim(input, split_dim), num_split, in __anondb9326b20802()
430 c->ReplaceDim(input, split_dim, split_dim_size, &out)); in __anondb9326b20802()
482 int64 split_dim = c->Value(split_dimension); in __anondb9326b20902() local
483 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anondb9326b20902()
502 TF_RETURN_IF_ERROR(c->ReplaceDim(input, split_dim, in __anondb9326b20902()
512 auto split_dim_size = c->Value(c->Dim(input, split_dim)); in __anondb9326b20902()
519 if (c->ValueKnown(c->Dim(input, split_dim)) && in __anondb9326b20902()
520 cumsum_outputs != c->Value(c->Dim(input, split_dim))) in __anondb9326b20902()
|
| D | ops.pbtxt | 27050 name: "split_dim"
27339 name: "split_dim"
27373 name: "split_dim"
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 105 auto split_dim = Const(scope_, 1, {}); in TEST_F() local
106 auto y = Split(scope_, split_dim, x, /* num_split */ 2); in TEST_F()
|
| /external/tensorflow/tensorflow/docs_src/install/ |
| D | migration.md | 255 * keyword argument `split_dim` should be renamed to `axis`
256 …_required=KeywordRequired(), sp_input=None, num_split=None, axis=None, name=None, split_dim=None)`.
258 * keyword argument `split_dim` should be renamed to `axis`
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lstm_layer_inference.pbtxt | 2630 name: "lstm_0/split/split_dim"
2654 input: "lstm_0/split/split_dim"
3284 name: "lstm_1/split/split_dim"
3308 input: "lstm_1/split/split_dim"
3938 name: "lstm_2/split/split_dim"
3962 input: "lstm_2/split/split_dim"
4592 name: "lstm_3/split/split_dim"
4616 input: "lstm_3/split/split_dim"
5246 name: "lstm_4/split/split_dim"
5270 input: "lstm_4/split/split_dim"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 624 split_dim=None): argument
670 split_dim)
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 675 const int split_dim = input_shape.dims(axis); in ProcessTensorFlowSplitOperator() local
676 CHECK_EQ(split_dim % op->num_split, 0); in ProcessTensorFlowSplitOperator()
677 const int split_depth = split_dim / op->num_split; in ProcessTensorFlowSplitOperator()
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| D | export_tensorflow.cc | 1007 const int split_dim = split_dim_data[0]; in ConvertSplitOperator() local
1008 CreateDummyConcatDimTensorConst(src_op.inputs[0], split_dim, in ConvertSplitOperator()
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.pbtxt | 1881 …'keyword_required\', \'sp_input\', \'num_split\', \'axis\', \'name\', \'split_dim\'], varargs=None…
|
| /external/tensorflow/ |
| D | RELEASE.md | 870 * `tf.sparse_split`: `split_dim` becomes `axis`
|