Searched refs:src_tensor_shape (Results 1 – 1 of 1) sorted by relevance
464 const TensorShape src_tensor_shape = src.shape(); in rearrange_data() local467 …const unsigned int O = src_tensor_shape[get_data_layout_dimension_index(data_layout, DataLayoutDi… in rearrange_data()468 …const unsigned int H = src_tensor_shape[get_data_layout_dimension_index(data_layout, DataLayoutDi… in rearrange_data()469 …const unsigned int W = src_tensor_shape[get_data_layout_dimension_index(data_layout, DataLayoutDi… in rearrange_data()470 …const unsigned int I = src_tensor_shape[get_data_layout_dimension_index(data_layout, DataLayoutDi… in rearrange_data()