Searched defs:input_strides (Results 1 – 11 of 11) sorted by relevance
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 130 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX8() local
245 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX16() local
360 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX32() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 39 const array<Index, NumDims>& input_strides, in GetInputIndex()
85 auto input_strides = internal::strides<Layout>(dims); in test_block_io_copy_data_from_source_to_target() local
158 auto input_strides = internal::strides<Layout>(dims); in test_block_io_copy_using_reordered_dimensions() local
|
| D | cxx11_tensor_block_access.cpp | 155 const array<Index, NumDims>& input_strides, in GetInputIndex()
199 const DSizes<StorageIndex, Dim> &input_dims, &input_strides, &output_dims, &output_strides; member in EqualityChecker
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 49 gtl::InlinedVector<int64, 8> input_strides(input_rank); in operator ()() local
|
| D | set_kernels.cc | 204 const VarDimArray& input_strides, in PopulateFromDenseGroup()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_transpose.cc | 59 int input_strides[4]; in Transpose() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 400 memory::dims input_dims, input_strides; in ComputeMklSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 961 const auto input_strides = ComputeStrides(input_lengths, input_literal); in ComputeFft() local
1306 const absl::Span<const int64> input_strides, in CopyDataFromInput()
1396 const absl::Span<const int64> input_strides, in CopyDataFromInput()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1210 mkldnn_dim_t* input_strides = new mkldnn_dim_t[kNumDims]; in CreateBlockedMemDescHelper() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2151 int input_strides[4]; in ShuffleArrayTemplate() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3394 std::vector<int64> input_strides = input_descriptor.vectorized_strides( in GetCudnnOperationGraph() local
3529 std::vector<int64> input_strides = input_descriptor.vectorized_strides( in GetCudnnFusedOperationGraph() local
|