| /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 TestX32() local
134 input_strides[i - 1] = input_stride; in TestX32()
164 i * input_strides[0] + j * input_strides[1] + k * input_strides[2] + in TestX32()
165 l * input_strides[3] + m * input_strides[4] + n * input_strides[5]; in TestX32()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 47 gtl::InlinedVector<int64, 8> input_strides(input_rank); in operator ()() local
49 input_strides[input_rank - 1] = 1; in operator ()()
51 input_strides[d] = input_strides[d + 1] * input_shape.dim_size(d + 1); in operator ()()
67 id += input_indices(i, j) * input_strides[j]; in operator ()()
|
| 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/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 400 memory::dims input_dims, input_strides; in ComputeMklSlice() local
420 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()
425 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()
428 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
435 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
|
| /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 | 2151 int input_strides[4]; in ShuffleArrayTemplate() local
2152 input_strides[3] = 1; in ShuffleArrayTemplate()
2153 input_strides[2] = extended_input_dims[3]; in ShuffleArrayTemplate()
2154 input_strides[1] = input_strides[2] * extended_input_dims[2]; in ShuffleArrayTemplate()
2155 input_strides[0] = input_strides[1] * extended_input_dims[1]; in ShuffleArrayTemplate()
2156 const int input_stride_0 = input_strides[extended_shuffle[3]]; in ShuffleArrayTemplate()
2157 const int input_stride_1 = input_strides[extended_shuffle[2]]; in ShuffleArrayTemplate()
2158 const int input_stride_2 = input_strides[extended_shuffle[1]]; in ShuffleArrayTemplate()
2159 const int input_stride_3 = input_strides[extended_shuffle[0]]; in ShuffleArrayTemplate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1135 const absl::Span<const int64> input_strides, in CopyDataFromInput() argument
1158 input_data[src_index + i * input_strides[axis]]); in CopyDataFromInput()
1167 GenerateIndices(fft_lengths, fft_strides, input_lengths, input_strides, in CopyDataFromInput()
1225 const absl::Span<const int64> input_strides, in CopyDataFromInput() argument
1231 fft_strides, input_lengths, input_strides, data); in CopyDataFromInput()
1235 fft_strides, input_lengths, input_strides, data); in CopyDataFromInput()
1412 const auto input_strides = ComputeStrides(input_lengths, input_literal); in HandleFft() local
1427 input_strides, absl::MakeSpan(data)); in HandleFft()
1441 input_strides, input_shape.rank(), 0, 0, base_case); in HandleFft()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1207 mkldnn_dim_t* input_strides = new mkldnn_dim_t[kNumDims]; in CreateBlockedMemDescHelper() local
1210 input_strides[i] = strides[i]; in CreateBlockedMemDescHelper()
1215 input_strides); in CreateBlockedMemDescHelper()
1217 delete[] input_strides; in CreateBlockedMemDescHelper()
1220 delete[] input_strides; in CreateBlockedMemDescHelper()
|