Searched refs:data_layout (Results 1 – 1 of 1) sorted by relevance
516 std::unique_ptr<uint8_t[]>& ptr_guard, bool data_layout) { in convertToNhwc() argument523 if (data_layout) { in convertToNhwc()560 bool data_layout, int* result) { in convertFromNhwc() argument565 if (data_layout) { in convertFromNhwc()871 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation() local878 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()883 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()884 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()887 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()920 if (data_layout) { in executeOperation()[all …]