| /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/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 130 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX8() local
134 input_strides[i - 1] = input_stride; in TestX8()
164 i * input_strides[0] + j * input_strides[1] + k * input_strides[2] + in TestX8()
165 l * input_strides[3] + m * input_strides[4] + n * input_strides[5]; in TestX8()
245 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX16() local
249 input_strides[i - 1] = input_stride; in TestX16()
279 i * input_strides[0] + j * input_strides[1] + k * input_strides[2] + in TestX16()
280 l * input_strides[3] + m * input_strides[4] + n * input_strides[5]; in TestX16()
360 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX32() local
364 input_strides[i - 1] = input_stride; in TestX32()
[all …]
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorTrace.h | 181 array<Index, NumInputDims> input_strides;
183 input_strides[0] = 1;
185 input_strides[i] = input_strides[i - 1] * input_dims[i - 1];
189 input_strides.back() = 1;
191 input_strides[i] = input_strides[i + 1] * input_dims[i + 1];
199 m_reducedStrides[reduced_index] = input_strides[i];
203 m_preservedStrides[output_index] = input_strides[i];
|
| D | TensorReduction.h | 591 array<Index, NumInputDims> input_strides;
593 input_strides[0] = 1;
595 input_strides[i] = input_strides[i-1] * input_dims[i-1];
598 input_strides.back() = 1;
600 input_strides[i] = input_strides[i + 1] * input_dims[i + 1];
608 m_reducedStrides[reduceIndex] = input_strides[i];
611 m_preservedStrides[outputIndex] = input_strides[i];
|
| D | TensorShuffling.h | 269 typename TensorBlockIO::Dimensions input_strides(m_unshuffledInputStrides);
270 TensorBlockIOSrc src(input_strides, m_impl.data(), srcCoeff(desc.offset()));
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 39 const array<Index, NumDims>& input_strides, in GetInputIndex() argument
45 input_index += idx * input_strides[output_to_input_dim_map[i]]; in GetInputIndex()
49 output_index * input_strides[output_to_input_dim_map[0]]; in GetInputIndex()
53 input_index += idx * input_strides[output_to_input_dim_map[i]]; in GetInputIndex()
57 output_index * input_strides[output_to_input_dim_map[NumDims - 1]]; in GetInputIndex()
85 auto input_strides = internal::strides<Layout>(dims); in test_block_io_copy_data_from_source_to_target() local
101 IOSrc src(input_strides, input_data, desc.offset()); in test_block_io_copy_data_from_source_to_target()
158 auto input_strides = internal::strides<Layout>(dims); in test_block_io_copy_using_reordered_dimensions() local
169 desc.offset(), output_to_input_dim_map, input_strides, in test_block_io_copy_using_reordered_dimensions()
184 IOSrc src(input_strides, input_data, first_coeff_index); in test_block_io_copy_using_reordered_dimensions()
[all …]
|
| D | cxx11_tensor_block_access.cpp | 155 const array<Index, NumDims>& input_strides, in GetInputIndex() argument
161 input_index += idx * input_strides[output_to_input_dim_map[i]]; in GetInputIndex()
165 output_index * input_strides[output_to_input_dim_map[0]]; in GetInputIndex()
169 input_index += idx * input_strides[output_to_input_dim_map[i]]; in GetInputIndex()
173 output_index * input_strides[output_to_input_dim_map[NumDims - 1]]; in GetInputIndex()
199 const DSizes<StorageIndex, Dim> &input_dims, &input_strides, &output_dims, &output_strides; member in EqualityChecker
210 …check_recursive(input + i % input_dims[depth] * input_strides[depth], output + i*output_strides[de… in check_recursive()
218 , input_dims(input_dims_), input_strides(input_strides_) in EqualityChecker()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 51 gtl::InlinedVector<int64_t, 8> input_strides(input_rank); in operator ()() local
53 input_strides[input_rank - 1] = 1; in operator ()()
55 input_strides[d] = input_strides[d + 1] * input_shape.dim_size(d + 1); in operator ()()
71 id += input_indices(i, j) * input_strides[j]; in operator ()()
|
| D | set_kernels.cc | 216 const VarDimArray& input_strides, in PopulateFromDenseGroup() argument
219 OP_REQUIRES(ctx, group_indices.size() == input_strides.size() - 1, in PopulateFromDenseGroup()
222 input_strides.size() - 1, ".")); in PopulateFromDenseGroup()
226 group_indices.begin(), group_indices.end(), input_strides.begin(), 0LL); in PopulateFromDenseGroup()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 407 memory::dims input_dims, input_strides; in ComputeMklSlice() local
427 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()
432 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()
435 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
442 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2153 int input_strides[4]; in ShuffleArrayTemplate() local
2154 input_strides[3] = 1; in ShuffleArrayTemplate()
2155 input_strides[2] = extended_input_dims[3]; in ShuffleArrayTemplate()
2156 input_strides[1] = input_strides[2] * extended_input_dims[2]; in ShuffleArrayTemplate()
2157 input_strides[0] = input_strides[1] * extended_input_dims[1]; in ShuffleArrayTemplate()
2158 const int input_stride_0 = input_strides[extended_shuffle[3]]; in ShuffleArrayTemplate()
2159 const int input_stride_1 = input_strides[extended_shuffle[2]]; in ShuffleArrayTemplate()
2160 const int input_stride_2 = input_strides[extended_shuffle[1]]; in ShuffleArrayTemplate()
2161 const int input_stride_3 = input_strides[extended_shuffle[0]]; in ShuffleArrayTemplate()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1244 dnnl_dim_t* input_strides = new dnnl_dim_t[kNumDims]; in CreateBlockedMemDescHelper() local
1247 input_strides[i] = strides[i]; in CreateBlockedMemDescHelper()
1252 input_strides); in CreateBlockedMemDescHelper()
1254 delete[] input_strides; in CreateBlockedMemDescHelper()
1257 delete[] input_strides; in CreateBlockedMemDescHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1643 const auto input_strides = ComputeStrides(input_lengths, input_literal); in ComputeFft() local
1658 input_lengths, input_strides, absl::MakeSpan(data)); in ComputeFft()
1672 input_strides, input_shape.rank(), 0, 0, base_case); in ComputeFft()
1988 const absl::Span<const int64_t> input_strides, in CopyDataFromInput() argument
2011 input_data[src_index + i * input_strides[axis]]); in CopyDataFromInput()
2020 GenerateIndices(fft_lengths, fft_strides, input_lengths, input_strides, in CopyDataFromInput()
2078 const absl::Span<const int64_t> input_strides, in CopyDataFromInput() argument
2084 input_strides, data); in CopyDataFromInput()
2088 input_lengths, input_strides, data); in CopyDataFromInput()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3521 std::vector<int64_t> input_strides = input_descriptor.vectorized_strides( in GetCudnnOperationGraph() local
3530 CreateCudnnTensor(input_dims, input_strides, 'x', in GetCudnnOperationGraph()
3656 std::vector<int64_t> input_strides = input_descriptor.vectorized_strides( in GetCudnnFusedOperationGraph() local
3665 CreateCudnnTensor(input_dims, input_strides, 'x', in GetCudnnFusedOperationGraph()
|