| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_concat_op.cc | 77 const int input_rank = input_shape.dims(); variable
79 ? input_rank + concat_dim_attr_
81 OP_REQUIRES(context, concat_dim >= 0 && concat_dim < input_rank,
83 -input_rank, ", ", input_rank,
88 context, current_shape.dims() == input_rank,
90 "Ranks of all input tensors must match: expected ", input_rank,
92 for (int j = 0; j < input_rank; ++j) {
112 gtl::InlinedVector<int64, 8> std_order(input_rank);
116 concat_order.reserve(input_rank);
[all...] |
| linalg_ops_common.cc | 118 int input_rank = -1; local
122 input_rank = in.dims();
124 context, input_rank >= 2,
126 " must have rank >= 2, got ", input_rank));
130 for (int dim = 0; dim < input_rank - 2; ++dim) {
135 OP_REQUIRES(context, input_rank == in.dims(),
138 for (int dim = 0; dim < input_rank - 2; ++dim) {
146 const int row_dimension = input_rank - 2;
147 const int col_dimension = input_rank - 1;
|
| reshape_util.cc | 50 const int64 input_rank = input_shape_in.NumElements(); local
109 gtl::InlinedVector<int64, 8> input_strides(input_rank);
110 input_strides[input_rank - 1] = 1;
111 for (int d = input_rank - 2; d >= 0; --d) {
130 for (int j = 0; j < input_rank; ++j) {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| depthtospace_op.cc | 45 int input_rank = input_tensor_shape.dims(); variable
47 OP_REQUIRES(ctx, kRequiredDims == input_rank,
49 "; got: ", input_rank));
56 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format_);
57 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format_);
62 reshaped_shape.reserve(input_rank);
63 transpose_order.reserve(input_rank);
64 output_shape.reserve(input_rank);
|
| spacetodepth_op.cc | 45 int input_rank = input_tensor_shape.dims(); variable
47 OP_REQUIRES(ctx, kRequiredDims == input_rank,
49 "; got ", input_rank));
56 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format_);
57 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format_);
62 reshaped_shape.reserve(input_rank);
63 transpose_order.reserve(input_rank);
64 output_shape.reserve(input_rank);
|
| batchtospace_op.cc | 28 const int input_rank = input_tensor_shape.dims(); local
34 ctx, input_rank >= 1 + block_rank,
36 " instead of ", input_rank));
71 std::vector<int64> reshaped_shape(input_rank + block_rank);
106 std::vector<int64> reshaped_permuted_shape(input_rank);
126 std::vector<int64> start_indices(input_rank, 0);
128 std::vector<int64> strides(input_rank, 1);
|
| spacetobatch_op.cc | 28 const int input_rank = input_tensor_shape.dims(); local
34 ctx, input_rank >= 1 + block_rank,
36 " instead of ", input_rank));
89 std::vector<int64> reshaped_padded_shape(input_rank + block_rank);
137 std::vector<int64> output_shape(input_rank);
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| layers.py | 198 input_rank = inputs.get_shape().ndims
200 if input_rank == 3:
202 input_rank)
203 elif input_rank == 4:
205 elif input_rank == 5:
207 input_rank)
210 input_rank)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| feature_column_ops.py | 45 1. If `output_rank > input_rank + 1`, raise a `ValueError`.
46 2. If `output_rank == input_rank + 1`, expand the tensor by one dimension.
47 3. If `output_rank == input_rank`, do nothing.
48 4. If `output_rank < input_rank`, flatten the inner dimensions of the tensor.
57 ValueError: if `output_rank > input_rank + 1` for the input tensor.
59 input_rank = tensor.get_shape().ndims
61 if input_rank is None and isinstance(tensor, sparse_tensor_py.SparseTensor):
63 input_rank = tensor.dense_shape.get_shape().as_list()[0]
65 if input_rank is None:
69 if output_rank > input_rank + 1
[all...] |
| layers.py | [all...] |
| feature_column.py | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| gather.cc | 87 const int input_rank = NumDimensions(input); local
90 GetTensorData<data_type>(input), GetTensorDims(input), input_rank, \
|
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference.cc | 857 int idx, int input_rank, DimensionHandle* out) {
866 if (input_rank < 0) {
869 } else if (val + input_rank < 0) {
871 val, " must be in range [-", input_rank,
872 ", ", input_rank, ")");
874 val += input_rank;
876 } else if (input_rank >= 0 && val >= input_rank) {
878 val, " must be in range [-", input_rank,
879 ", ", input_rank, ")");
[all...] |
| common_shape_fns.cc | 1090 const int32 input_rank = c->Rank(input); local
[all...] |
| shape_inference.h | 499 Status MakeDimForScalarInputWithNegativeIndexing(int idx, int input_rank,
|
| /external/tensorflow/tensorflow/core/ops/ |
| math_ops.cc | 791 const int32 input_rank = c->Rank(input_shape); local
792 if (input_rank <= 1) {
802 std::vector<DimensionHandle> dims(input_rank - 1);
818 int64 axis = dimension_val < 0 ? dimension_val + input_rank : dimension_val;
819 if (axis < 0 || axis >= input_rank) {
821 "Dimension (", dimension_val, ") must be in the range [", -input_rank,
822 ", ", input_rank, "), where ", input_rank,
828 for (int i = 0; i < input_rank; ++i) {
[all...] |
| array_ops.cc | 74 // Paddings is a matrix of [input_rank, 2].
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| spectral_ops.py | 109 input_rank = _array_ops.rank(input_tensor)
111 outer_dims = _math_ops.maximum(0, input_rank - fft_rank)
|
| nn_ops.py | [all...] |
| array_grad.py | 246 input_rank = array_ops.rank(input_vec)
249 shape = array_ops.stack([input_rank, 1])
|
| math_ops.py | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.cc | 488 int input_rank = c->Rank(c->input(0)); local
489 Tensor t(node->output_type(0), TensorShape({input_rank}));
492 for (int i = 0; i < input_rank; i++) {
503 for (int i = 0; i < input_rank; i++) {
516 int32 input_rank = c->Rank(c->input(0)); local
518 t.flat<int32>()(0) = input_rank;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util.cc | [all...] |
| /external/tensorflow/tensorflow/cc/gradients/ |
| math_grad.cc | 627 // input_rank = 4
628 auto input_rank = Size(scope, input_shape); local
632 auto axes = Mod(scope, Add(scope, reduction_axes, input_rank), input_rank);
634 // This [0..input_rank) range of integers is used in DynamicStitch to
637 auto input_rank_range = Range(scope, zero, input_rank, one);
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| model.h | 1315 int input_rank = 0; member in struct:toco::GatherOperator
[all...] |
