Searched refs:input_strides_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op.h | 138 input_strides_[0] = 1;
141 input_strides_[i + 1] = input_strides_[i] * input_dims[i];
145 input_strides_[numext::maxi(0, Dims - 1)] = 1;
148 input_strides_[i - 1] = input_strides_[i] * input_dims[i];
178 return helper(coords, input_strides_, impl_);
301 input_index += ToInputCoord(coords[dim], dim) * input_strides_[dim];
312 input_index += ToInputCoord(k, dim) * input_strides_[dim];
319 input_index += ToInputCoord(k, dim) * input_strides_[dim];
330 array<Index, Dims> input_strides_;
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_access.cpp | 215 … const DSizes<StorageIndex, Dim> &input_dims_, const DSizes<StorageIndex, Dim> &input_strides_, in EqualityChecker() argument
218 , input_dims(input_dims_), input_strides(input_strides_) in EqualityChecker()
|