| /external/tensorflow/tensorflow/cc/framework/ |
| gradient_checker_test.cc | 129 auto split_dim = Const(scope, 1, {}); local
130 auto y = Split(scope, split_dim, x, /* num_split */ 2);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| split_op.cc | 57 int32 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() variable
59 OP_REQUIRES(ctx, 0 <= split_dim && split_dim < input_shape.dims(),
61 ") <= split_dim < input rank (",
71 ctx, input_shape.dim_size(split_dim) % num_split == 0,
74 "dimension, but got split_dim ",
75 split_dim_orig, " (size = ", input_shape.dim_size(split_dim), ") ",
80 const int32 slice_size = input_shape.dim_size(split_dim) / num_split;
99 begin[split_dim] = i * slice_size;
100 limits[split_dim] = (i + 1) * slice_size
118 int32 split_dim; variable
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_split_op.cc | 33 const int64 split_dim = context->input(0).scalar<int64>()(); variable
53 input_shape.dim_size(0) && split_dim < input_shape.vec<int64>().size(),
55 "Input split_dim should be between 0 and rank (",
56 input_shape.vec<int64>().size(), "), got ", split_dim));
60 num_split_ >= 1 && num_split_ <= input_shape.vec<int64>()(split_dim),
63 input_shape.vec<int64>()(split_dim), "), got ",
69 sparse::SparseTensor::Split<T>(sparse_tensor, split_dim, num_split_);
|
| split_op.cc | 53 const int32 split_dim = local
58 context, 0 <= split_dim && split_dim < input_shape.dims(),
60 ") <= split_dim < input rank (", input.dims(),
68 OP_REQUIRES(context, input_shape.dim_size(split_dim) % num_split == 0,
71 "dimension, but got split_dim ",
72 split_dim, " (size = ", input_shape.dim_size(split_dim),
90 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) {
103 const TensorShape& input_shape, int32 split_dim) const
[all...] |
| split_v_op.cc | 60 const int32 split_dim = local
84 context, 0 <= split_dim && split_dim < input.dims(),
86 ") <= split_dim < input rank (", input.dims(),
89 Tlen input_size_split_dim = input_shape.dim_size(split_dim);
125 "input shape along split_dim exactly if "
127 "the input along split_dim if not fully "
143 if ((split_dim == 0) && IsInnerDimsSizeAligned<T>(input_shape)) {
157 const TensorShape& input_shape, const int32 split_dim) const {
161 for (int i = 0; i < split_dim; ++i)
195 const int32 split_dim = variable
307 const int32 split_dim = variable
[all...] |
| /external/tensorflow/tensorflow/cc/gradients/ |
| array_grad_test.cc | 105 auto split_dim = Const(scope_, 1, {}); local
106 auto y = Split(scope_, split_dim, x, /* num_split */ 2);
|
| /external/tensorflow/tensorflow/core/ops/ |
| array_ops_test.cc | 1107 Tensor split_dim = test::AsTensor<int32>({1, 2}); local
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| propagate_fixed_sizes.cc | 675 const int split_dim = input_shape.dims(axis); local
676 CHECK_EQ(split_dim % op->num_split, 0);
677 const int split_depth = split_dim / op->num_split;
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| export_tensorflow.cc | 1007 const int split_dim = split_dim_data[0]; local
[all...] |
