Searched refs:kOHWI (Results 1 – 4 of 4) sorted by relevance
2020 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()2026 output_axes_order == AxesOrder::kOHWI) { in GetShuffleShape()2030 } else if (input_axes_order == AxesOrder::kOHWI && in GetShuffleShape()2170 case AxesOrder::kOHWI: in AxesCount()
184 kOHWI, // Our standard for conv weights enumerator
402 ConvertFloatTensorConst(model, weights_array_name, AxesOrder::kOHWI, in ConvertConvOperator()555 ConvertFloatTensorConst(model, weights_array_name, AxesOrder::kOHWI, in ConvertTransposeConvOperator()
646 reorder->output_axes_order = AxesOrder::kOHWI; in ConvertConvOperator()