Home
last modified time | relevance | path

Searched refs:data_layout (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/common/
DCpuExecutor.cpp518 std::unique_ptr<uint8_t[]>& ptr_guard, bool data_layout) { in convertToNhwc() argument
525 if (data_layout) { in convertToNhwc()
562 bool data_layout, int* result) { in convertFromNhwc() argument
567 if (data_layout) { in convertFromNhwc()
873 bool data_layout = inCount == 3 ? getScalarData<bool>(operands[ins[2]]) : false; in executeOperation() local
880 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
885 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
886 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
889 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
922 if (data_layout) { in executeOperation()
[all …]