Searched refs:result_dtypes (Results 1 – 1 of 1) sorted by relevance
214 std::vector<DataType> result_dtypes(ctx->num_outputs()); in CompileSingleOp() local215 for (int i = 0; i < result_dtypes.size(); ++i) { in CompileSingleOp()216 result_dtypes[i] = ctx->expected_output_dtype(i); in CompileSingleOp()219 args, result_dtypes, result); in CompileSingleOp()