Lines Matching refs:output
802 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
803 Shape outShape = output.shape(); in executeOperation()
808 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
809 copyData(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
820 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
821 Shape outShape = output.shape(); in executeOperation()
830 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
831 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
834 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
870 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
884 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
885 Shape outShape = output.shape(); in executeOperation()
894 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
895 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
899 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
935 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
946 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation() local
952 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && lookup.Eval(); in executeOperation()
962 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation() local
970 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && in executeOperation()
974 RunTimeOperandInfo& output = operands[outs[LSHProjection::kOutputTensor]]; in executeOperation() local
977 !setInfoAndAllocateIfNeeded(&output, outputShape, &result)) { in executeOperation()
1045 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation() local
1055 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && lstm_cell.Eval(); in executeOperation()
1061 RunTimeOperandInfo& output = operands[outs[Multinomial::kOutputTensor]]; in executeOperation() local
1067 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && in executeOperation()
1076 RunTimeOperandInfo& output = operands[outs[RNN::kOutputTensor]]; in executeOperation() local
1083 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && rnn_cell.Eval(); in executeOperation()
1087 RunTimeOperandInfo& output = operands[outs[SVDF::kOutputTensor]]; in executeOperation() local
1094 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && svdf.Eval(); in executeOperation()
1105 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1106 Shape outShape = output.shape(); in executeOperation()
1115 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1116 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1122 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1162 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1177 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1178 Shape outShape = output.shape(); in executeOperation()
1187 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1188 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1195 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1239 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1253 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1254 Shape outShape = output.shape(); in executeOperation()
1258 !setInfoAndAllocateIfNeeded(&output, outShape, &result)) { in executeOperation()
1265 reinterpret_cast<float*>(output.buffer), outShape); in executeOperation()
1271 reinterpret_cast<_Float16*>(output.buffer), outShape); in executeOperation()
1277 output.buffer, outShape); in executeOperation()
1283 output.buffer, outShape); in executeOperation()
1292 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1293 Shape outShape = output.shape(); in executeOperation()
1296 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1297 cast::eval(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
1307 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1308 Shape outShape = output.shape(); in executeOperation()
1312 !setInfoAndAllocateIfNeeded(&output, outShape, &result)) { in executeOperation()
1318 keepDims > 0, reinterpret_cast<_Float16*>(output.buffer), in executeOperation()
1324 reinterpret_cast<float*>(output.buffer), outShape); in executeOperation()
1329 reinterpret_cast<uint8_t*>(output.buffer), outShape); in executeOperation()
1334 reinterpret_cast<int8_t*>(output.buffer), outShape); in executeOperation()
1345 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1346 Shape outShape = output.shape(); in executeOperation()
1350 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1351 argMinMaxGeneric(input.buffer, input.shape(), axis, isArgMin, output.buffer, in executeOperation()
1361 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1362 Shape outShape = output.shape(); in executeOperation()
1365 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1366 expand_dims::eval(input.buffer, input.shape(), axis, output.buffer, outShape); in executeOperation()
1451 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1452 Shape outputShape = output.shape(); in executeOperation()
1456 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && in executeOperation()
1458 isMinimum, output.buffer, outputShape); in executeOperation()
1496 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1497 Shape outShape = output.shape(); in executeOperation()
1506 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1507 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1526 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1592 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1604 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1605 Shape outShape = output.shape(); in executeOperation()
1610 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1612 reinterpret_cast<const int32_t*>(multiples.buffer), output.buffer, in executeOperation()
1622 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation() local
1630 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && in executeOperation()
1640 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation() local
1641 Shape outShape = output.shape(); in executeOperation()
1644 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1646 output.buffer, outShape); in executeOperation()