Searched refs:input_strides (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_transpose.cc | 59 int input_strides[4]; in Transpose() local
60 input_strides[3] = 1; in Transpose()
61 input_strides[2] = extended_input_dims[3]; in Transpose()
62 input_strides[1] = input_strides[2] * extended_input_dims[2]; in Transpose()
63 input_strides[0] = input_strides[1] * extended_input_dims[1]; in Transpose()
64 const int input_stride_0 = input_strides[extended_perm[3]]; in Transpose()
65 const int input_stride_1 = input_strides[extended_perm[2]]; in Transpose()
66 const int input_stride_2 = input_strides[extended_perm[1]]; in Transpose()
67 const int input_stride_3 = input_strides[extended_perm[0]]; in Transpose()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 109 gtl::InlinedVector<int64, 8> input_strides(input_rank); in Reshape() local
111 input_strides[input_rank - 1] = 1; in Reshape()
113 input_strides[d] = input_strides[d + 1] * input_shape.dim_size(d + 1); in Reshape()
135 id += input_ind(i, j) * input_strides[j]; in Reshape()
|
| D | mkl_slice_op.cc | 405 memory::dims input_dims, input_strides; in ComputeMklSlice() local
407 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()
410 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
|
| D | set_kernels.cc | 204 const VarDimArray& input_strides, in PopulateFromDenseGroup() argument
207 OP_REQUIRES(ctx, group_indices.size() == input_strides.size() - 1, in PopulateFromDenseGroup()
210 input_strides.size() - 1, ".")); in PopulateFromDenseGroup()
214 group_indices.begin(), group_indices.end(), input_strides.begin(), 0LL); in PopulateFromDenseGroup()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReduction.h | 448 array<Index, NumInputDims> input_strides;
450 input_strides[0] = 1;
452 input_strides[i] = input_strides[i-1] * input_dims[i-1];
455 input_strides.back() = 1;
457 input_strides[i] = input_strides[i + 1] * input_dims[i + 1];
465 m_reducedStrides[reduceIndex] = input_strides[i];
468 m_preservedStrides[outputIndex] = input_strides[i];
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2107 int input_strides[4]; in ShuffleArrayTemplate() local
2108 input_strides[3] = 1; in ShuffleArrayTemplate()
2109 input_strides[2] = extended_input_dims[3]; in ShuffleArrayTemplate()
2110 input_strides[1] = input_strides[2] * extended_input_dims[2]; in ShuffleArrayTemplate()
2111 input_strides[0] = input_strides[1] * extended_input_dims[1]; in ShuffleArrayTemplate()
2112 const int input_stride_0 = input_strides[extended_shuffle[3]]; in ShuffleArrayTemplate()
2113 const int input_stride_1 = input_strides[extended_shuffle[2]]; in ShuffleArrayTemplate()
2114 const int input_stride_2 = input_strides[extended_shuffle[1]]; in ShuffleArrayTemplate()
2115 const int input_stride_3 = input_strides[extended_shuffle[0]]; in ShuffleArrayTemplate()
|