Searched refs:output_to_input_dim_map (Results 1 – 2 of 2) sorted by relevance
38 const array<Index, NumDims>& output_to_input_dim_map, in GetInputIndex() argument45 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()135 DSizes<Index, NumDims> output_to_input_dim_map; in test_block_io_copy_using_reordered_dimensions() local139 output_to_input_dim_map[shuffle[i]] = i; in test_block_io_copy_using_reordered_dimensions()169 desc.offset(), output_to_input_dim_map, input_strides, in test_block_io_copy_using_reordered_dimensions()189 dim_map[j] = static_cast<int>(output_to_input_dim_map[j]); in test_block_io_copy_using_reordered_dimensions()197 dst_dims[output_to_input_dim_map[out_dim]] = blk_dims[out_dim]; in test_block_io_copy_using_reordered_dimensions()
154 const array<Index, NumDims>& output_to_input_dim_map, in GetInputIndex() argument161 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()